diff --git a/setup.py b/setup.py
index c01f667..677bb4f 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@ setup(
     name='EasyGA',
     version='1.2.0',
     description='EasyGA is a python package designed to provide an easy-to-use Genetic Algorithm. The package is designed to work right out of the box, while also allowing the user to customize features as they see fit.',
-    py_modules=["EasyGA","attributes","test_EasyGA","decorators.py"],
+    py_modules=["EasyGA","attributes","test_EasyGA","decorators"],
     packages=find_packages(where='EasyGA'),
     package_dir={
         '': 'EasyGA',