diff --git a/qt/applications/workbench/workbench/utils/test/test_io.py b/qt/applications/workbench/workbench/utils/test/test_io.py
index 10c159e0458715763bcc90258d15eb33049a9fc2..af97e7cd6a02692e8ba3c88af2b7d4f7c14a1c81 100644
--- a/qt/applications/workbench/workbench/utils/test/test_io.py
+++ b/qt/applications/workbench/workbench/utils/test/test_io.py
@@ -8,7 +8,7 @@
 from unittest import TestCase
 from unittest.mock import patch
 
-from PyQt5.QtWidgets import QInputDialog
+from qtpy.QtWidgets import QInputDialog
 from workbench.utils.io import input_qinputdialog