diff --git a/main.py b/main.py
index 18483c3..9e52045 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,7 @@
 from world_maker.world_maker import *
 
 def main():
-    world_maker()
+    rectangle_house_mountain, rectangle_building, skeleton_highway, skeleton_mountain = world_maker()
 
 if __name__ == '__main__':
-    rectangle_house_mountain,rectangle_building, skeleton_highway, skeleton_mountain = main()
\ No newline at end of file
+    main()
\ No newline at end of file