diff --git a/tests/test_application.py b/tests/test_application.py
index baf5d34..495df13 100644
--- a/tests/test_application.py
+++ b/tests/test_application.py
@@ -1,4 +1,4 @@
-from app.main import sum_two
+from app import main
 
 def test_hello_thing():
     result = sum_two(1, 2)