diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 62eb5e9ecdc1765a4258e7329416a9d3df4edefe..51e378afa3ab55cd6ae5c21353a1e8473463e8d2 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -206,11 +206,12 @@ setup_pybind11(cfg)
         except Exception as e:
             print(e)
             torch_extension = load(hash,
-                               [file_name],
-                               with_cuda=self.is_cuda,
-                               extra_cflags=['--std=c++14', get_compiler_config()['flags'].replace('--std=c++11', '')],
-                               extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
-                               build_directory=cache_dir,
-                               extra_include_paths=[get_pycuda_include_path(),
-                                                    get_pystencils_include_path()])
+                                   [file_name],
+                                   with_cuda=self.is_cuda,
+                                   extra_cflags=['--std=c++14',
+                                                 get_compiler_config()['flags'].replace('--std=c++11', '')],
+                                   extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
+                                   build_directory=cache_dir,
+                                   extra_include_paths=[get_pycuda_include_path(),
+                                                        get_pystencils_include_path()])
         return torch_extension