diff --git a/__init__.py b/__init__.py
index 35309a7..a1d140c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -16,6 +16,10 @@ import os
 
 from .tree import *
 
+# Set the `geometry_script` module to the current module in case the folder is named differently.
+import sys
+sys.modules['geometry_script'] = sys.modules[__name__]
+
 bl_info = {
     "name" : "Geometry Script",
     "author" : "Carson Katri",
diff --git a/examples/Cube.py b/examples/Repeat Grid.py
similarity index 100%
rename from examples/Cube.py
rename to examples/Repeat Grid.py
diff --git a/examples/repeat.py b/examples/repeat.py
deleted file mode 100644
index f6df584..0000000
--- a/examples/repeat.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from geometry_script import *
-
-@tree
-def repeat(size: Vector):
-    cubes = []
-    for x in range(5):
-        for y in range(5):
-            for z in range(5):
-                c = cube(size=(0.1, 0.1, 0.1))
-                cubes.append(c.transform(translation=(x, y, z)))
-    return join_geometry(geometry=cubes)
\ No newline at end of file