diff --git a/src/sakia/tests/functional/wot_tab/test_wot_tab.py b/src/sakia/tests/functional/wot_tab/test_wot_tab.py
index 0bac059cb2e89515ed55bf521c5d3247026eeb85..9f4f3d9915f238b4b08baba1224707cf3bbe60a6 100644
--- a/src/sakia/tests/functional/wot_tab/test_wot_tab.py
+++ b/src/sakia/tests/functional/wot_tab/test_wot_tab.py
@@ -58,7 +58,7 @@ class TestWotTab(unittest.TestCase, QuamashTest):
         future = asyncio.Future()
 
         def open_widget():
-            wot_tab.show()
+            wot_tab.widget.show()
             return future
 
         async def async_open_widget():
@@ -68,13 +68,13 @@ class TestWotTab(unittest.TestCase, QuamashTest):
             await open_widget()
 
         def close_dialog():
-            if wot_tab.isVisible():
-                wot_tab.close()
+            if wot_tab.wiget.isVisible():
+                wot_tab.widget.close()
             future.set_result(True)
 
         async def exec_test():
             await asyncio.sleep(1)
-            self.assertTrue(wot_tab.isVisible())
+            self.assertTrue(wot_tab.widget.isVisible())
             self.lp.call_soon(close_dialog)
 
         asyncio.ensure_future(exec_test())
diff --git a/src/sakia/tests/unit/gui/test_context_menu.py b/src/sakia/tests/unit/gui/test_context_menu.py
index 2e3fa7bb3c58b2743db718bf6afcacbfb726ac2b..299918431949ab12364406a421775b5fde4de434 100644
--- a/src/sakia/tests/unit/gui/test_context_menu.py
+++ b/src/sakia/tests/unit/gui/test_context_menu.py
@@ -1,11 +1,10 @@
 import unittest
 from unittest.mock import patch, MagicMock, Mock
 from asynctest.mock import CoroutineMock
-from asynctest.mock import MagicMock as AsyncMagicMock
-from PyQt5.QtCore import QLocale, pyqtSignal
+from PyQt5.QtCore import QLocale
 from sakia.tests import QuamashTest
 from sakia.tests.mocks.bma import nice_blockchain
-from sakia.gui.widgets import ContextMenu
+from sakia.gui.widgets.context_menu import ContextMenu
 
 
 class TestContextMenu(unittest.TestCase, QuamashTest):