diff --git a/tests/ui/test_main_ui.py b/tests/ui/test_main_ui.py
index dcca9cb..49f3277 100644
--- a/tests/ui/test_main_ui.py
+++ b/tests/ui/test_main_ui.py
@@ -1,6 +1,6 @@
 from PyQt5 import QtCore
 import pytest
-import joystick_diagrams.main as ui
+import joystick_diagrams.__main__ as ui
 import joystick_diagrams.version as version