diff --git a/lib/ssn/pair_wise_distance_cuda_source.py b/lib/ssn/pair_wise_distance_cuda_source.py
index 49aa127..51d0629 100644
--- a/lib/ssn/pair_wise_distance_cuda_source.py
+++ b/lib/ssn/pair_wise_distance_cuda_source.py
@@ -11,7 +11,6 @@
 #include <ATen/core/TensorAccessor.h>
 #include <ATen/cuda/CUDAContext.h>
 
-#include <THC/THC.h>
 #include <THC/THCAtomics.cuh>
 #include <THC/THCDeviceUtils.cuh>
 
@@ -174,4 +173,4 @@
   m.def("forward", &forward_cuda, "pair_wise_distance forward");
   m.def("backward", &backward_cuda, "pair_wise_distance backward");
 }
-'''
\ No newline at end of file
+'''