diff --git a/setup.py b/setup.py
index 1aa3687..0bfd2f3 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="quickai",  # Replace with your own username
-    version="1.3.5",
+    version="1.3.6",
     author="geekjr",
     author_email="author@example.com",
     description="QuickAI is a Python library that makes it extremely easy to experiment with state-of-the-art "
@@ -21,7 +21,7 @@
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-    packages=[".quickai"],
+    packages=setuptools.find_packages(),
     python_requires=">=3.6",
     install_requires=[
         'scikit-learn',