diff --git a/tests/test_hello.py b/tests/test_hello.py
index 252565fd617c319ab0ba84bb0b8eb084bd924364..f86e70701c22d59a318f56cfd49e4360820b247c 100644
--- a/tests/test_hello.py
+++ b/tests/test_hello.py
@@ -1,9 +1,9 @@
 import unittest
-from pak.hello import hello
+import pak
 
 class TestMyModule(unittest.TestCase):
     def test_hello(self):
-        self.assertEqual(hello(), "Hello, World!")
+        self.assertEqual(pak.main(), "Hello, World!")
 
 if __name__ == '__main__':
     unittest.main()
\ No newline at end of file