diff --git a/dan/ocr/manager/training.py b/dan/ocr/manager/training.py
index edb906aadd053f209da1d15888afddf3299ff192..1034bb65b598e3aeaf7043091fe3b3e3eb179da8 100644
--- a/dan/ocr/manager/training.py
+++ b/dan/ocr/manager/training.py
@@ -903,15 +903,10 @@ class GenericTrainingManager:
         return info_dict
 
 
-class OCRManager(GenericTrainingManager):
-    def __init__(self, params):
-        super(OCRManager, self).__init__(params)
-        self.params["model"]["vocab_size"] = len(self.dataset.charset)
-
-
-class Manager(OCRManager):
+class Manager(GenericTrainingManager):
     def __init__(self, params):
         super(Manager, self).__init__(params)
+        self.params["model"]["vocab_size"] = len(self.dataset.charset)
 
     def load_save_info(self, info_dict):
         if "curriculum_config" in info_dict: