diff --git a/04_mnist_basics.ipynb b/04_mnist_basics.ipynb
index 675bb5b..31436e9 100644
--- a/04_mnist_basics.ipynb
+++ b/04_mnist_basics.ipynb
@@ -4825,10 +4825,10 @@
     "class BasicOptim:\n",
     "    def __init__(self,params,lr): self.params,self.lr = list(params),lr\n",
     "\n",
-    "    def step(self, *args, **kwargs):\n",
+    "    def step(self):\n",
     "        for p in self.params: p.data -= p.grad.data * self.lr\n",
     "\n",
-    "    def zero_grad(self, *args, **kwargs):\n",
+    "    def zero_grad(self):\n",
     "        for p in self.params: p.grad = None"
    ]
   },