diff --git a/mithril/backends/with_manualgrad/numpy_backend/backend.py b/mithril/backends/with_manualgrad/numpy_backend/backend.py
index 006c30ed..986460bd 100644
--- a/mithril/backends/with_manualgrad/numpy_backend/backend.py
+++ b/mithril/backends/with_manualgrad/numpy_backend/backend.py
@@ -146,7 +146,6 @@ def ones(
     def argmax(
         self,
         input: np.ndarray[Any, Any],
-        *,
         axis: int | None = None,
         keepdim: bool = False,
         cache: CacheType | None = None,