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