diff --git a/frigate/test/test_custom_classification.py b/frigate/test/test_custom_classification.py
index 3a0644c48..a554c5dba 100644
--- a/frigate/test/test_custom_classification.py
+++ b/frigate/test/test_custom_classification.py
@@ -26,11 +26,8 @@ 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):
     """Test that zone information is correctly added to custom classification MQTT messages"""