diff --git a/codebert.py b/codebert.py
index 2761f97..d31d66e 100644
--- a/codebert.py
+++ b/codebert.py
@@ -30,7 +30,7 @@ log = logging.getLogger()
 
 
 @register(output_schema=['vec'])
-class AutoTransformers(NNOperator):
+class CodeBert(NNOperator):
     """
     An operator generates an embedding for code or natural language text
     using a pretrained codebert model gathered by huggingface.