diff --git a/tests/test_extension.py b/tests/test_extension.py
index 8dd51f6..35fc285 100644
--- a/tests/test_extension.py
+++ b/tests/test_extension.py
@@ -2,7 +2,7 @@ from __future__ import unicode_literals
 
 import unittest
 
-from mopidy_touchscreen import Extension, frontend as frontend_lib
+from mopidy_touchscreen import Extension, touch_screen as frontend_lib
 
 
 class ExtensionTest(unittest.TestCase):
@@ -24,4 +24,4 @@ class ExtensionTest(unittest.TestCase):
         #self.assertIn('username', schema)
         #self.assertIn('password', schema)
 
-    # TODO Write more tests
\ No newline at end of file
+    # TODO Write more tests