diff --git a/pytec/pyproject.toml b/pytec/pyproject.toml
index 73ce527..3eeb969 100644
--- a/pytec/pyproject.toml
+++ b/pytec/pyproject.toml
@@ -14,8 +14,6 @@ license = {text = "GPLv3"}
 tec_qt = "tec_qt:main"
 
 [tool.setuptools]
-packages.find = {}
-py-modules = ["aioexample", "autotune", "example", "plot", "tec_qt", "ui_tec_qt", "waitingspinnerwidget"]
-
-[tool.setuptools.package-data]
-"*" = ["*.*"]
+packages.find = {} # Use setuptools custom discovery, package directory structure isn't standard
+package-data = {"*" = ["*.ico", "*.ui", "*.json"]}
+py-modules = ["aioexample", "autotune", "example", "plot", "tec_qt"]
diff --git a/pytec/setup.py b/pytec/setup.py
index 4ac8f58..0229d0d 100644
--- a/pytec/setup.py
+++ b/pytec/setup.py
@@ -14,5 +14,6 @@ setup(
             "tec_qt = tec_qt:main",
         ]
     },
-    py_modules=['tec_qt', 'ui_tec_qt', 'autotune', 'waitingspinnerwidget'],
+    py_modules=['aioexample', 'autotune', 'example', 'plot', 'tec_qt'],
+    package_data={"*": ["*.ico", "*.ui", "*.json"]},
 )