diff --git a/source/module_base/module_container/ATen/ops/test/CMakeLists.txt b/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
index 260aaa800f..0cc2c090ca 100644
--- a/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
+++ b/source/module_base/module_container/ATen/ops/test/CMakeLists.txt
@@ -1,7 +1,7 @@
 AddTest(
   TARGET container_ops_uts
   LIBS parameter ${math_libs} 
-  SOURCES einsum_op_test.cpp linalg_op_test.cpp
+  SOURCES einsum_op_test.cpp linalg_op_test.cpp ../../kernels/lapack.cpp
 )
 
 target_link_libraries(container_ops_uts container base device)
\ No newline at end of file
diff --git a/source/module_elecstate/test/CMakeLists.txt b/source/module_elecstate/test/CMakeLists.txt
index 6a10494d4d..350e89153f 100644
--- a/source/module_elecstate/test/CMakeLists.txt
+++ b/source/module_elecstate/test/CMakeLists.txt
@@ -33,7 +33,7 @@ AddTest(
 AddTest(
   TARGET elecstate_fp_energy
   LIBS parameter  ${math_libs} base device
-  SOURCES elecstate_fp_energy_test.cpp ../fp_energy.cpp
+  SOURCES elecstate_fp_energy_test.cpp ../fp_energy.cpp ../elecstate_energy_terms.cpp
 )
 
 AddTest(
@@ -64,7 +64,7 @@ AddTest(
 AddTest(
   TARGET elecstate_energy
   LIBS parameter  ${math_libs} base device 
-  SOURCES elecstate_energy_test.cpp ../elecstate_energy.cpp ../fp_energy.cpp
+  SOURCES elecstate_energy_test.cpp ../elecstate_energy.cpp ../fp_energy.cpp ../elecstate_energy_terms.cpp
 )
 
 AddTest(
diff --git a/source/module_elecstate/test/elecstate_base_test.cpp b/source/module_elecstate/test/elecstate_base_test.cpp
index 02905c64ed..cbbd1aac68 100644
--- a/source/module_elecstate/test/elecstate_base_test.cpp
+++ b/source/module_elecstate/test/elecstate_base_test.cpp
@@ -1,5 +1,7 @@
 #include <string>
 
+#include"module_cell/setup_nonlocal.h"
+
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #define private public
diff --git a/source/module_elecstate/test/elecstate_print_test.cpp b/source/module_elecstate/test/elecstate_print_test.cpp
index 0fb0248e64..dfb7e94559 100644
--- a/source/module_elecstate/test/elecstate_print_test.cpp
+++ b/source/module_elecstate/test/elecstate_print_test.cpp
@@ -1,5 +1,7 @@
 #include <string>
 
+#include"module_cell/setup_nonlocal.h"
+
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #define private public