diff --git a/include/flexflow/model.h b/include/flexflow/model.h
index 494b8ee04a..702a3311ad 100644
--- a/include/flexflow/model.h
+++ b/include/flexflow/model.h
@@ -713,7 +713,7 @@ class FFModel {
       const Tensor input,
       int embed_dim,
       int num_heads,
-                                      int num_hidden_layers,
+      int num_hidden_layers,
       int kdim = 0,
       int vdim = 0,
       float dropout = 0.0f,