diff --git a/tests/functional/openlp_core/lib/test_theme.py b/tests/functional/openlp_core/lib/test_theme.py
index e7f935918..e294e5214 100644
--- a/tests/functional/openlp_core/lib/test_theme.py
+++ b/tests/functional/openlp_core/lib/test_theme.py
@@ -25,7 +25,7 @@ Package to test the openlp.core.lib.theme package.
 from pathlib import Path
 from unittest import TestCase
 
-from openlp.core.lib.theme import BackgroundType
+from openlp.core.lib.theme import BackgroundType, Theme
 
 
 class TestBackgroundType(TestCase):