diff --git a/manim_rubikscube/cubie.py b/manim_rubikscube/cubie.py
index 131147d..b230071 100644
--- a/manim_rubikscube/cubie.py
+++ b/manim_rubikscube/cubie.py
@@ -2,7 +2,7 @@
 from manim.constants import *
 from manim.utils.color import *
 from manim.utils.space_ops import z_to_vector
-from manim.mobject.geometry import Square
+from manim.mobject.geometry.polygram import Square
 from .cube_utils import get_faces_of_cubie
 
 import numpy as np
@@ -84,4 +84,4 @@ def init_colors(self):
             width=self.background_stroke_width,
             opacity=self.background_stroke_opacity,
             family=False
-        )
\ No newline at end of file
+        )