diff --git a/ultravox/training/train.py b/ultravox/training/train.py
index 52be715..5281d80 100644
--- a/ultravox/training/train.py
+++ b/ultravox/training/train.py
@@ -139,8 +139,8 @@ def train(args: config_base.TrainConfig):
         model = ultravox_model.UltravoxModel(config)
 
     assert model.get_input_embeddings().num_embeddings == len(
-         text_tokenizer
-     ), f"Model and tokenizer mismatch: {model.get_input_embeddings().num_embeddings} != {len(text_tokenizer)}"
+        text_tokenizer
+    ), f"Model and tokenizer mismatch: {model.get_input_embeddings().num_embeddings} != {len(text_tokenizer)}"
 
     model.language_model.config.use_cache = False
     if args.disable_layerdrop and hasattr(model.audio_tower.config, "layerdrop"):