diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py
index 6570c5d31e..ddd0308379 100644
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@@ -40,7 +40,7 @@ class ExampleModel(RootModule):
     # ---------------------
     # TRAINING
     # ---------------------
-    def forward(self, x, a):
+    def forward(self, x):
 
         x = self.c_d1(x)
         x = F.tanh(x)