diff --git a/torchmdnet/neighbors/__init__.py b/torchmdnet/neighbors/__init__.py
index 7b9c52336..3f4fdc29d 100644
--- a/torchmdnet/neighbors/__init__.py
+++ b/torchmdnet/neighbors/__init__.py
@@ -5,7 +5,7 @@
 def compile_extension():
     src_dir = os.path.dirname(__file__)
     sources = ["neighbors.cpp", "neighbors_cpu.cpp"] + (
-        ["neighbors_cuda.cu", "backwards.cu"] if torch.cuda.is_available() else []
+        ["neighbors_cuda.cu"] if torch.cuda.is_available() else []
     )
     sources = [os.path.join(src_dir, name) for name in sources]
     cpp_extension.load(