diff --git a/statapp/generate_window.py b/statapp/generate_window.py
index e299ccf..7785bcd 100644
--- a/statapp/generate_window.py
+++ b/statapp/generate_window.py
@@ -1,6 +1,5 @@
-from PySide6.QtCore import Slot
-from PySide6.QtGui import QIntValidator
-from PySide6.QtWidgets import QDialog
+from PySide2.QtCore import Slot
+from PySide2.QtWidgets import QDialog
 
 from statapp.ui.ui_generate_window import Ui_GenerateWindow
 
diff --git a/statapp/main_window.py b/statapp/main_window.py
index 01d135c..9f1a161 100644
--- a/statapp/main_window.py
+++ b/statapp/main_window.py
@@ -1,3 +1,4 @@
+import numpy as np
 from PySide2.QtCore import Slot
 from PySide2.QtWidgets import QMainWindow
 
diff --git a/statapp/models/data_model.py b/statapp/models/data_model.py
index f24ff5d..a46f31b 100644
--- a/statapp/models/data_model.py
+++ b/statapp/models/data_model.py
@@ -1,6 +1,6 @@
 import numpy as np
-from PySide6 import QtCore
-from PySide6.QtCore import Qt
+from PySide2 import QtCore
+from PySide2.QtCore import Qt
 
 
 class DataModel(QtCore.QAbstractTableModel):
diff --git a/statapp/ui/ui_generate_window.py b/statapp/ui/ui_generate_window.py
index 1f8d7db..076f949 100644
--- a/statapp/ui/ui_generate_window.py
+++ b/statapp/ui/ui_generate_window.py
@@ -3,21 +3,15 @@
 ################################################################################
 ## Form generated from reading UI file 'generate_window.ui'
 ##
-## Created by: Qt User Interface Compiler version 6.5.2
+## Created by: Qt User Interface Compiler version 5.15.2
 ##
 ## WARNING! All changes made in this file will be lost when recompiling UI file!
 ################################################################################
 
-from PySide6.QtCore import (QCoreApplication, QDate, QDateTime, QLocale,
-    QMetaObject, QObject, QPoint, QRect,
-    QSize, QTime, QUrl, Qt)
-from PySide6.QtGui import (QBrush, QColor, QConicalGradient, QCursor,
-    QFont, QFontDatabase, QGradient, QIcon,
-    QImage, QKeySequence, QLinearGradient, QPainter,
-    QPalette, QPixmap, QRadialGradient, QTransform)
-from PySide6.QtWidgets import (QApplication, QDialog, QDoubleSpinBox, QGridLayout,
-    QLabel, QPushButton, QSizePolicy, QSpinBox,
-    QWidget)
+from PySide2.QtCore import *
+from PySide2.QtGui import *
+from PySide2.QtWidgets import *
+
 
 class Ui_GenerateWindow(object):
     def setupUi(self, GenerateWindow):