diff --git a/nemo/collections/llm/bert/model/base.py b/nemo/collections/llm/bert/model/base.py
index c55ae1108845..8c11dfc0500c 100644
--- a/nemo/collections/llm/bert/model/base.py
+++ b/nemo/collections/llm/bert/model/base.py
@@ -48,7 +48,7 @@
     from megatron.core.models.bert.bert_model import BertModel as MCoreBert
 except (ImportError, ModuleNotFoundError) as e:
     HAVE_TE = False
-    MCoreBert = TransformerLayer # Place holder for import checking. BERT requires TE installed.
+    MCoreBert = TransformerLayer  # Place holder for import checking. BERT requires TE installed.
 
 if TYPE_CHECKING:
     from nemo.collections.common.tokenizers.tokenizer_spec import TokenizerSpec