diff --git a/setup.py b/setup.py
index 81faa3d8..f56277ad 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@ def set_torch_cuda_arch_list():
 ExtensionType = CppExtension if not use_cuda else CUDAExtension
 extensions = ExtensionType(
     name='torchmdnet.extensions.torchmdnet_extensions',
-    sources=[os.path.join(extension_root, "extensions.cpp")] + neighbor_sources,
+    sources=[os.path.join(extension_root, "torchmdnet_extensions.cpp")] + neighbor_sources,
     include_dirs=include_paths(),
     define_macros=[('WITH_CUDA', 1)] if use_cuda else [],
 )
diff --git a/torchmdnet/extensions/extensions.cpp b/torchmdnet/extensions/torchmdnet_extensions.cpp
similarity index 100%
rename from torchmdnet/extensions/extensions.cpp
rename to torchmdnet/extensions/torchmdnet_extensions.cpp