diff --git a/source/source_pw/module_pwdft/op_pw_exx_pot.cpp b/source/source_pw/module_pwdft/op_pw_exx_pot.cpp
index fd76372e317..2a99833bbfa 100644
--- a/source/source_pw/module_pwdft/op_pw_exx_pot.cpp
+++ b/source/source_pw/module_pwdft/op_pw_exx_pot.cpp
@@ -25,7 +25,7 @@ void get_exx_potential(const K_Vectors* kv,
                        const CoulombParam& coulomb_param_in)
 {
     using setmem_real_cpu_op = base_device::memory::set_memory_op<Real, base_device::DEVICE_CPU>;
-    using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op<Real, base_device::DEVICE_CPU, Device>;
+    using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op<Real, Device, base_device::DEVICE_CPU>;
 
     Real nqs_half1 = 0.5 * kv->nmp[0];
     Real nqs_half2 = 0.5 * kv->nmp[1];
@@ -238,7 +238,7 @@ void get_exx_stress_potential(const K_Vectors* kv,
                               const CoulombParam& coulomb_param_in)
 {
     using setmem_real_cpu_op = base_device::memory::set_memory_op<Real, base_device::DEVICE_CPU>;
-    using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op<Real, base_device::DEVICE_CPU, Device>;
+    using syncmem_real_c2d_op = base_device::memory::synchronize_memory_op<Real, Device, base_device::DEVICE_CPU>;
 
     Real nqs_half1 = 0.5 * kv->nmp[0];
     Real nqs_half2 = 0.5 * kv->nmp[1];
diff --git a/tests/01_PW/CASES_GPU.txt b/tests/01_PW/CASES_GPU.txt
index 32a694ab0af..80e9078f9c8 100644
--- a/tests/01_PW/CASES_GPU.txt
+++ b/tests/01_PW/CASES_GPU.txt
@@ -97,8 +97,8 @@ scf_out_elf
 095_PW_NPT
 #096_PW_NVT
 #097_PW_PBE0
-#097_PW_PBE0_AFM
-#097_PW_PBE0_FM
+097_PW_PBE0_AFM
+097_PW_PBE0_FM
 099_PW_15_SO_avg
 #100_PW_DJ_SO
 #101_PW_W90