diff --git a/setup.py b/setup.py
index 4559689a6..dfd722b1d 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
-  version = '0.7.0',
+  version = '0.7.1',
   license='MIT',
   description = 'Swarms - Pytorch',
   author = 'Kye Gomez',
diff --git a/swarms/__init__.py b/swarms/__init__.py
index 46071ccef..a93f3df79 100644
--- a/swarms/__init__.py
+++ b/swarms/__init__.py
@@ -1 +1,2 @@
-from swarms import Swarms, swarm
\ No newline at end of file
+# from swarms import Swarms, swarm
+from swarms.swarms import Swarms, swarm
\ No newline at end of file