diff --git a/src/pyronn_torch/codegen.py b/src/pyronn_torch/codegen.py
index 5f1e96e8bf7666d342333660a03c8e1d5dacf5b9..97b37632e755efd5ba3ce0ae6b9db16bfb97449b 100644
--- a/src/pyronn_torch/codegen.py
+++ b/src/pyronn_torch/codegen.py
@@ -154,10 +154,11 @@ try:
                                                    const float volume_origin_x, const float volume_origin_y,
                                                    const int detector_size, const float detector_spacing, const float detector_origin);
     """),  # noqa
-            }
+}
 except Exception as e:
     import warnings
     warnings.warn(str(e))
+    TorchModule = None
 
 
 def get_pyronn_cuda_kernels():