diff --git a/tests/test_main.py b/tests/test_main.py
index c747c461..bbd64adf 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,4 +1,5 @@
 import pytest
+import re
 
 from app.main import count_occurrences
 
@@ -22,10 +23,10 @@ def test_count_occurrences(phrase, letter, count):
 
     
 def test_removed_comment():
-import app
-with open(app.main.__file__, "r") as f:
-    file_content = f.read()
-    comment = re.compile("# write your code here")
-    assert not comment.search(
-        file_content
-    ), "You have to remove the unnecessary comment '# write your code here'"
+    import app
+    with open(app.main.__file__, "r") as f:
+        file_content = f.read()
+        comment = re.compile("# write your code here")
+        assert not comment.search(
+            file_content
+        ), "You have to remove the unnecessary comment '# write your code here'"