diff --git a/.idea/AutoFramework.iml b/.idea/AutoFramework.iml
index 485bc5b..f7f2049 100644
--- a/.idea/AutoFramework.iml
+++ b/.idea/AutoFramework.iml
@@ -2,6 +2,8 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/Base" isTestSource="true" />
+      <sourceFolder url="file://$MODULE_DIR$/Business" isTestSource="true" />
       <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
     <orderEntry type="jdk" jdkName="Python 3.10 (AutoFramework)" jdkType="Python SDK" />
diff --git a/main.py b/main.py
index e0369f7..e904651 100644
--- a/main.py
+++ b/main.py
@@ -1 +1 @@
-print('test')
\ No newline at end of file
+print('test')