diff --git a/README.md b/README.md
index 2a90df5..5130258 100644
--- a/README.md
+++ b/README.md
@@ -14,7 +14,8 @@ The implementation is based on the architectures of [DeepLabV3+](https://github.
 ### Gaussian PSF Layer
 In order to build the Gaussian PSF layer on your own machine, run the following line
 ```
-python ext/setup.py install
+cd ext
+python setup.py install
 ```
 
 ## Datasets
diff --git a/Utils/GaussPSFLayer.py b/Utils/GaussPSFLayer.py
index 5f976a4..8878adf 100644
--- a/Utils/GaussPSFLayer.py
+++ b/Utils/GaussPSFLayer.py
@@ -2,7 +2,7 @@
 import torch.nn as nn
 from torch.autograd import Function
 
-import gauss_psf
+import gauss_psf_cuda
 
 
 class GaussPSFFunction(Function):
diff --git a/ext/setup.py b/ext/setup.py
index 81e774d..b2d084a 100644
--- a/ext/setup.py
+++ b/ext/setup.py
@@ -6,9 +6,9 @@
     ext_modules=[
         CUDAExtension('gauss_psf_cuda', [
             'gauss_psf_cuda.cpp',
-            'gauss_psf_kernel.cu',
+            'gauss_psf_cuda_kernel.cu',
         ]),
     ],
     cmdclass={
         'build_ext': BuildExtension
-    })
\ No newline at end of file
+    })