diff --git a/frigate/test/test_custom_classification.py b/frigate/test/test_custom_classification.py
index a554c5dba..830fb0628 100644
--- a/frigate/test/test_custom_classification.py
+++ b/frigate/test/test_custom_classification.py
@@ -26,7 +26,7 @@ MOCK_MODULES = [
     "tensorflow.lite.python.interpreter",
 ]
 
-ORIGINAL_MODULES = {mod: sys.modules[mod] for mod in MOCK_MODULES}}
+ORIGINAL_MODULES = {mod: sys.modules[mod] for mod in MOCK_MODULES}
 
 
 class TestCustomObjectClassificationZones(unittest.TestCase):