diff --git a/qt/pe/details_dialog.py b/qt/pe/details_dialog.py
index eb480815..07ecdfcb 100644
--- a/qt/pe/details_dialog.py
+++ b/qt/pe/details_dialog.py
@@ -12,10 +12,7 @@ from hscommon.trans import trget
 from ..details_dialog import DetailsDialog as DetailsDialogBase
 from ..details_table import DetailsTable
 from .image_viewer import (
-    ViewerToolBar, QWidgetImageViewer,
-    ScrollAreaImageViewer, GraphicsViewViewer,
-    QWidgetController, ScrollAreaController,
-    GraphicsViewController)
+    ViewerToolBar, ScrollAreaImageViewer, ScrollAreaController)
 tr = trget("ui")
 
 
@@ -49,12 +46,7 @@ class DetailsDialog(DetailsDialogBase):
         self.selectedImageViewer = ScrollAreaImageViewer(self, "selectedImage")
         self.horizontalLayout.addWidget(self.selectedImageViewer, 0, 0, 3, 1)
         # Use a specific type of controller depending on the underlying viewer type
-        if isinstance(self.selectedImageViewer, QWidgetImageViewer):
-            self.vController = QWidgetController(self)
-        elif isinstance(self.selectedImageViewer, ScrollAreaImageViewer):
-            self.vController = ScrollAreaController(self)
-        elif isinstance(self.selectedImageViewer, GraphicsViewViewer):
-            self.vController = GraphicsViewController(self)
+        self.vController = ScrollAreaController(self)
 
         self.verticalToolBar = ViewerToolBar(self, self.vController)
         self.verticalToolBar.setOrientation(Qt.Orientation(Qt.Vertical))
diff --git a/qt/pe/image_viewer.py b/qt/pe/image_viewer.py
index 677071e0..54366e98 100644
--- a/qt/pe/image_viewer.py
+++ b/qt/pe/image_viewer.py
@@ -2,7 +2,8 @@
 # which should be included with this package. The terms are also available at
 # http://www.gnu.org/licenses/gpl-3.0.html
 
-from PyQt5.QtCore import QObject, Qt, QSize, QRectF, QPointF, QPoint, pyqtSlot, pyqtSignal, QEvent
+from PyQt5.QtCore import (
+    QObject, Qt, QSize, QRectF, QPointF, QPoint, pyqtSlot, pyqtSignal, QEvent)
 from PyQt5.QtGui import QPixmap, QPainter, QPalette, QCursor, QIcon, QKeySequence
 from PyQt5.QtWidgets import (
     QGraphicsView, QGraphicsScene, QGraphicsPixmapItem,