diff --git a/tests/functional/openlp_core_lib/test_screen.py b/tests/functional/openlp_core_lib/test_screen.py
index 4164607de..0b708dd94 100644
--- a/tests/functional/openlp_core_lib/test_screen.py
+++ b/tests/functional/openlp_core_lib/test_screen.py
@@ -4,6 +4,7 @@ Package to test the openlp.core.lib.screenlist package.
 import copy
 from unittest import TestCase
 
+from mock import MagicMock
 from PyQt4 import QtGui, QtCore
 
 from openlp.core.lib import ScreenList