diff --git a/deluge/tests/test_ui_entry.py b/deluge/tests/test_ui_entry.py
index 8a032a26c..ef28cc4ad 100644
--- a/deluge/tests/test_ui_entry.py
+++ b/deluge/tests/test_ui_entry.py
@@ -165,10 +165,10 @@ class DelugeEntryTestCase(BaseTestCase):
 class GtkUIBaseTestCase(UIBaseTestCase):
     """Implement all GtkUI tests here"""
 
-    def test_start_gtkui(self):
+    def test_start_gtk3ui(self):
         self.patch(sys, 'argv', utf8_encode_structure(self.var['sys_arg_cmd']))
 
-        from deluge.ui.gtkui import gtkui
+        from deluge.ui.gtk3 import gtkui
 
         with mock.patch.object(gtkui.GtkUI, 'start', autospec=True):
             self.exec_command()
@@ -196,10 +196,10 @@ class GtkUIScriptEntryTestCase(BaseTestCase, GtkUIBaseTestCase):
     def __init__(self, testname):
         super(GtkUIScriptEntryTestCase, self).__init__(testname)
         GtkUIBaseTestCase.__init__(self)
-        from deluge.ui import gtkui
+        from deluge.ui import gtk3
 
         self.var['cmd_name'] = 'deluge-gtk'
-        self.var['start_cmd'] = gtkui.start
+        self.var['start_cmd'] = gtk3.start
         self.var['sys_arg_cmd'] = ['./deluge-gtk']
 
     def set_up(self):