diff --git a/src/polyA_classifier.py b/src/polyA_classifier.py
index 428d5b5c2f54c841fbea2e41d3eb4ba7d08a9076..4570e9d9b70b7e0bd582b2073cd61f15090672e7 100644
--- a/src/polyA_classifier.py
+++ b/src/polyA_classifier.py
@@ -49,7 +49,7 @@ class PolyAClassifier:
         'C': 1.0
     }
 
-    def __init__(self, model = Net, state_dict_path: str = './models/internal_priming.pth'):
+    def __init__(self, model=Net, state_dict_path: str = './models/internal_priming.pth'):
         """Returns a stateless classifier with the model loaded.
 
         Args: