diff --git a/ranger21/ranger21.py b/ranger21/ranger21.py
index ab29073..1a529b6 100644
--- a/ranger21/ranger21.py
+++ b/ranger21/ranger21.py
@@ -512,7 +512,7 @@ class Ranger21(TO.Optimizer):
 
         loss = None
         if closure is not None and isinstance(closure, collections.Callable):
-            with torch.grad():
+            with torch.enable_grad():
                 loss = closure()
 
         param_size = 0