diff --git a/source/module_esolver/esolver_ks.h b/source/module_esolver/esolver_ks.h
index 323b60316b..61e8fd3f29 100644
--- a/source/module_esolver/esolver_ks.h
+++ b/source/module_esolver/esolver_ks.h
@@ -5,7 +5,7 @@
 #include "module_cell/klist.h"
 #include "module_elecstate/module_charge/charge_mixing.h"
 #include "module_hamilt_general/hamilt.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hsolver/hsolver.h"
 #include "module_io/cal_test.h"
 #include "module_psi/psi.h"
diff --git a/source/module_hamilt_pw/hamilt_pwdft/CMakeLists.txt b/source/module_hamilt_pw/hamilt_pwdft/CMakeLists.txt
index eb76d7dc74..18e6518a8d 100644
--- a/source/module_hamilt_pw/hamilt_pwdft/CMakeLists.txt
+++ b/source/module_hamilt_pw/hamilt_pwdft/CMakeLists.txt
@@ -26,8 +26,6 @@ list(APPEND objects
     VL_in_pw.cpp
     VNL_in_pw.cpp
     VNL_grad_pw.cpp
-    wavefunc.cpp
-    wf_atomic.cpp
     structure_factor.cpp
     structure_factor_k.cpp
     soc.cpp
diff --git a/source/module_hamilt_pw/hamilt_pwdft/VNL_in_pw.cpp b/source/module_hamilt_pw/hamilt_pwdft/VNL_in_pw.cpp
index 1b3f92b60a..dcf70385e6 100644
--- a/source/module_hamilt_pw/hamilt_pwdft/VNL_in_pw.cpp
+++ b/source/module_hamilt_pw/hamilt_pwdft/VNL_in_pw.cpp
@@ -13,7 +13,7 @@
 #include "module_base/timer.h"
 #include "module_hamilt_pw/hamilt_pwdft/global.h"
 #include "module_hamilt_pw/hamilt_pwdft/kernels/vnl_op.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 
 pseudopot_cell_vnl::pseudopot_cell_vnl()
 {
diff --git a/source/module_hsolver/hsolver.h b/source/module_hsolver/hsolver.h
index 593b0526c2..9a624d664f 100644
--- a/source/module_hsolver/hsolver.h
+++ b/source/module_hsolver/hsolver.h
@@ -4,7 +4,7 @@
 #include "module_base/macros.h"
 #include "module_elecstate/elecstate.h"
 #include "module_hamilt_general/hamilt.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hamilt_pw/hamilt_stodft/sto_wf.h"
 #include "module_psi/psi.h"
 
diff --git a/source/module_hsolver/hsolver_lcaopw.cpp b/source/module_hsolver/hsolver_lcaopw.cpp
index be509b5629..6c1169df1a 100644
--- a/source/module_hsolver/hsolver_lcaopw.cpp
+++ b/source/module_hsolver/hsolver_lcaopw.cpp
@@ -7,7 +7,7 @@
 #include "module_elecstate/elecstate_pw.h"
 #include "module_hamilt_pw/hamilt_pwdft/global.h"
 #include "module_hamilt_pw/hamilt_pwdft/hamilt_pw.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hsolver/diago_iter_assist.h"
 #include "module_parameter/parameter.h"
 
diff --git a/source/module_hsolver/hsolver_pw.cpp b/source/module_hsolver/hsolver_pw.cpp
index 05a4a893eb..977fc262ed 100644
--- a/source/module_hsolver/hsolver_pw.cpp
+++ b/source/module_hsolver/hsolver_pw.cpp
@@ -5,7 +5,7 @@
 #include "module_base/tool_quit.h"
 #include "module_elecstate/elecstate_pw.h"
 #include "module_hamilt_general/hamilt.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hsolver/diag_comm_info.h"
 #include "module_hsolver/diago_bpcg.h"
 #include "module_hsolver/diago_cg.h"
diff --git a/source/module_hsolver/hsolver_pw.h b/source/module_hsolver/hsolver_pw.h
index dd71cdf367..f7724961c1 100644
--- a/source/module_hsolver/hsolver_pw.h
+++ b/source/module_hsolver/hsolver_pw.h
@@ -4,7 +4,7 @@
 #include "hsolver.h"
 #include "module_base/macros.h"
 #include "module_basis/module_pw/pw_basis_k.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 
 namespace hsolver
 {
diff --git a/source/module_hsolver/test/hsolver_pw_sup.h b/source/module_hsolver/test/hsolver_pw_sup.h
index 6b9d872e34..594777072b 100644
--- a/source/module_hsolver/test/hsolver_pw_sup.h
+++ b/source/module_hsolver/test/hsolver_pw_sup.h
@@ -183,7 +183,7 @@ template class DiagoIterAssist<std::complex<double>, base_device::DEVICE_CPU>;
 
 } // namespace hsolver
 
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 namespace hamilt {
 
 template <>
diff --git a/source/module_hsolver/test/hsolver_supplementary_mock.h b/source/module_hsolver/test/hsolver_supplementary_mock.h
index 59e07df85f..6f7b0df338 100644
--- a/source/module_hsolver/test/hsolver_supplementary_mock.h
+++ b/source/module_hsolver/test/hsolver_supplementary_mock.h
@@ -1,6 +1,6 @@
 #pragma once
 #include "module_elecstate/elecstate.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 
 namespace elecstate
 {
diff --git a/source/module_io/test/for_testing_input_conv.h b/source/module_io/test/for_testing_input_conv.h
index 941656af3e..8ec1791995 100644
--- a/source/module_io/test/for_testing_input_conv.h
+++ b/source/module_io/test/for_testing_input_conv.h
@@ -16,7 +16,7 @@
 #include "module_hamilt_lcao/module_tddft/td_velocity.h"
 #include "module_hamilt_pw/hamilt_pwdft/VNL_in_pw.h"
 #include "module_hamilt_pw/hamilt_pwdft/structure_factor.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hsolver/hsolver_lcao.h"
 #include "module_io/berryphase.h"
 #include "module_io/restart.h"
diff --git a/source/module_io/write_dos_lcao.cpp b/source/module_io/write_dos_lcao.cpp
index dcdfc211f0..89dda2e591 100644
--- a/source/module_io/write_dos_lcao.cpp
+++ b/source/module_io/write_dos_lcao.cpp
@@ -8,7 +8,7 @@
 #include "module_base/global_function.h"
 #include "module_base/global_variable.h"
 #include "module_hamilt_pw/hamilt_pwdft/global.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "write_orb_info.h"
 
 #include <sys/time.h>
diff --git a/source/module_psi/CMakeLists.txt b/source/module_psi/CMakeLists.txt
index 7fbef390f4..77afc701dc 100644
--- a/source/module_psi/CMakeLists.txt
+++ b/source/module_psi/CMakeLists.txt
@@ -8,6 +8,8 @@ add_library(
     psi_overall_init
     OBJECT
     psi_init.cpp
+    wavefunc.cpp
+    wf_atomic.cpp
 )
 
 add_library(
diff --git a/source/module_psi/psi_init.h b/source/module_psi/psi_init.h
index 077422abbb..a631939dfe 100644
--- a/source/module_psi/psi_init.h
+++ b/source/module_psi/psi_init.h
@@ -1,7 +1,7 @@
 #ifndef W_ABACUS_DEVELOP_ABACUS_DEVELOP_SOURCE_MODULE_HAMILT_PW_HAMILT_PWDFT_WFINIT_H
 #define W_ABACUS_DEVELOP_ABACUS_DEVELOP_SOURCE_MODULE_HAMILT_PW_HAMILT_PWDFT_WFINIT_H
 #include "module_hamilt_general/hamilt.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_psi/psi_initializer.h"
 
 namespace psi
diff --git a/source/module_hamilt_pw/hamilt_pwdft/wavefunc.cpp b/source/module_psi/wavefunc.cpp
similarity index 100%
rename from source/module_hamilt_pw/hamilt_pwdft/wavefunc.cpp
rename to source/module_psi/wavefunc.cpp
diff --git a/source/module_hamilt_pw/hamilt_pwdft/wavefunc.h b/source/module_psi/wavefunc.h
similarity index 98%
rename from source/module_hamilt_pw/hamilt_pwdft/wavefunc.h
rename to source/module_psi/wavefunc.h
index c1d2386582..44ca10a849 100644
--- a/source/module_hamilt_pw/hamilt_pwdft/wavefunc.h
+++ b/source/module_psi/wavefunc.h
@@ -6,7 +6,7 @@
 #include "module_base/global_variable.h"
 #include "module_base/matrix.h"
 #include "module_hamilt_general/hamilt.h"
-#include "wf_atomic.h"
+#include "module_psi/wf_atomic.h"
 
 class wavefunc : public WF_atomic
 {
diff --git a/source/module_hamilt_pw/hamilt_pwdft/wf_atomic.cpp b/source/module_psi/wf_atomic.cpp
similarity index 99%
rename from source/module_hamilt_pw/hamilt_pwdft/wf_atomic.cpp
rename to source/module_psi/wf_atomic.cpp
index ed213c198d..484fc3b847 100644
--- a/source/module_hamilt_pw/hamilt_pwdft/wf_atomic.cpp
+++ b/source/module_psi/wf_atomic.cpp
@@ -5,10 +5,11 @@
 #include "module_base/math_sphbes.h"
 #include "module_base/math_polyint.h"
 #include "module_base/math_ylmreal.h"
-#include "module_hamilt_pw/hamilt_pwdft/soc.h"
-#include <complex>
 #include "module_base/timer.h"
 #include "module_base/tool_quit.h"
+#include "module_hamilt_pw/hamilt_pwdft/soc.h"
+
+#include <complex>
 
 WF_atomic::WF_atomic()
 {
diff --git a/source/module_hamilt_pw/hamilt_pwdft/wf_atomic.h b/source/module_psi/wf_atomic.h
similarity index 96%
rename from source/module_hamilt_pw/hamilt_pwdft/wf_atomic.h
rename to source/module_psi/wf_atomic.h
index d827f4903c..babd02ffa8 100644
--- a/source/module_hamilt_pw/hamilt_pwdft/wf_atomic.h
+++ b/source/module_psi/wf_atomic.h
@@ -6,8 +6,9 @@
 #include "module_base/global_variable.h"
 #include "module_base/realarray.h"
 #include "module_basis/module_pw/pw_basis_k.h"
+#include "module_hamilt_pw/hamilt_pwdft/structure_factor.h"
+
 #include "module_psi/psi.h"
-#include "structure_factor.h"
 
 class WF_atomic
 {
@@ -33,7 +34,7 @@ class WF_atomic
 
     void init_at_1(Structure_Factor *sf_in); // from init_at_1.f90
 
-    void print_PAOs(void)const;
+    void print_PAOs()const;
 
     public: //template change to public, will be refactor later. added by zhengdy 20230302
     int *irindex = nullptr;
diff --git a/source/module_ri/exx_lip.hpp b/source/module_ri/exx_lip.hpp
index f2ea0f79f4..94689f5167 100644
--- a/source/module_ri/exx_lip.hpp
+++ b/source/module_ri/exx_lip.hpp
@@ -13,7 +13,7 @@
 #include "module_base/vector3.h"
 #include "module_hamilt_pw/hamilt_pwdft/global.h"
 #include "module_cell/klist.h"
-#include "module_hamilt_pw/hamilt_pwdft/wavefunc.h"
+#include "module_psi/wavefunc.h"
 #include "module_hamilt_lcao/hamilt_lcaodft/wavefunc_in_pw.h"
 #include "module_base/lapack_connector.h"
 #include "module_base/parallel_global.h"