diff --git a/extensions/csrc/cuda/pybind/layer_norm.cpp b/extensions/csrc/cuda/pybind/layer_norm.cpp
index 3439e5e71..b1f7c2543 100644
--- a/extensions/csrc/cuda/pybind/layer_norm.cpp
+++ b/extensions/csrc/cuda/pybind/layer_norm.cpp
@@ -7,7 +7,7 @@
 #include <cassert>
 #include <vector>
 
-#include "../common/micros.h"
+#include "../../common/micros.h"
 
 namespace {
 
diff --git a/extensions/moe/moe_cuda.py b/extensions/moe/moe_cuda.py
index 722daae33..7a4744d4d 100644
--- a/extensions/moe/moe_cuda.py
+++ b/extensions/moe/moe_cuda.py
@@ -11,7 +11,7 @@ class MoeCudaExtension(_CudaExtension):
         return ret
 
     def sources_files(self):
-        ret = [self.csrc_abs_path(fname) for fname in ["cuda/moe.cpp", "cuda/moe_kernel.cu"]]
+        ret = [self.csrc_abs_path(fname) for fname in ["cuda/pybind/moe.cpp", "cuda/moe_kernel.cu"]]
         return ret
 
     def cxx_flags(self):