diff --git a/mopidy_touchscreen/screens/search_screen.py b/mopidy_touchscreen/screens/search_screen.py
index 38305e7..1cd0df1 100644
--- a/mopidy_touchscreen/screens/search_screen.py
+++ b/mopidy_touchscreen/screens/search_screen.py
@@ -1,3 +1,5 @@
+import pygame
+
 from base_screen import BaseScreen
 
 from ..input import InputManager