diff --git a/src/ui/screens/import_screen.py b/src/ui/screens/import_screen.py
index f2ffaee..53eb6d2 100644
--- a/src/ui/screens/import_screen.py
+++ b/src/ui/screens/import_screen.py
@@ -79,8 +79,9 @@ class ImportScreen(QWidget):
                 Qt.AspectRatioMode.KeepAspectRatio,
                 Qt.TransformationMode.SmoothTransformation
             )
-            btn.setIcon(scaled)
+            from PyQt6.QtGui import QIcon
             from PyQt6.QtCore import QSize
+            btn.setIcon(QIcon(scaled))
             btn.setIconSize(QSize(170, 170))
 
         btn.clicked.connect(lambda checked, p=path: self.main_window.show_editor(p))