diff --git a/tests.py b/tests.py
index 0f6ea94..61eaf45 100644
--- a/tests.py
+++ b/tests.py
@@ -1,2 +1,4 @@
 def test_initial_layout(snap_compare):  # pyright: ignore[reportMissingParameterType, reportUnknownParameterType]
-    assert snap_compare("src/teilchensammler_cli/main.py", terminal_size=(130, 40))
+    from teilchensammler_cli.main import app
+
+    assert snap_compare(app, terminal_size=(130, 40))