diff --git a/src/__init__.py b/src/__init__.py
index f7a3e3f..cc33d2f 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1,3 +1,3 @@
 import EasyGA
-import running_test.py
-import test_EasyGA.py
+import running_test
+import test_EasyGA