diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py
index 59fb6830..7dacf6ea 100644
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@@ -201,7 +201,7 @@ class ExampleModel(RootModule):
         # network params
         parser.opt_list('--drop_prob', default=0.2, options=[0.2, 0.5], type=float, tunable=False)
         parser.add_argument('--in_features', default=28*28)
-        parser.add_argument('--hidden_dim', default=500)
+        parser.add_argument('--hidden_dim', default=50000)
         parser.add_argument('--out_features', default=10)
 
         # data