diff --git a/demo/dj_integration.ipynb b/demo/dj_integration.ipynb
index 686a6ce..5a7ef04 100644
--- a/demo/dj_integration.ipynb
+++ b/demo/dj_integration.ipynb
@@ -141,7 +141,7 @@
     "method_fn = \"mei.methods.gradient_ascent\"\n",
     "method_config = dict(\n",
     "    initial=dict(path=\"mei.initial.RandomNormal\"),\n",
-    "    optimizer=dict(path=\"torch.optim.SGD\", kwargs=dict(lr=0.1)),\n",
+    "    optimizer=dict(path=\"torch.optim.SGD\", kwargs=dict(lr=1)),\n",
     "    stopper=dict(path=\"mei.stoppers.NumIterations\", kwargs=dict(num_iterations=1000)),\n",
     "    objectives=[dict(path=\"mei.objectives.EvaluationObjective\", kwargs=dict(interval=10))],\n",
     "    device=\"cuda\",\n",