diff --git a/src/pystencils_autodiff/backends/_torch_native.py b/src/pystencils_autodiff/backends/_torch_native.py
index bba20bc7417c41765f16c10ea6193cfdf7245b25..c9c78ddc143fdf1b57bbfb495d4bbeac677d0e93 100644
--- a/src/pystencils_autodiff/backends/_torch_native.py
+++ b/src/pystencils_autodiff/backends/_torch_native.py
@@ -64,7 +64,7 @@ def create_autograd_function(autodiff_obj, use_cuda):
                         field.shape,
                         dtype=numpy_dtype_to_torch(field.dtype.numpy_dtype),
                         device=next(chain(args, kwargs.values())).device)
-                except:
+                except Exception:
                     shape = next(filter(lambda x: isinstance(x, torch.Tensor), chain(args, kwargs.values()))).shape
                     kwargs[field.name] = torch.zeros(
                         shape,