diff --git a/deepmd/pd/utils/nlist.py b/deepmd/pd/utils/nlist.py
index 923d43cbfd..58d02b8aec 100644
--- a/deepmd/pd/utils/nlist.py
+++ b/deepmd/pd/utils/nlist.py
@@ -385,7 +385,11 @@ def build_multiple_neighbor_list(
         ).to(device=nlist.place)
         # nb x nloc x nsel
         nlist = paddle.concat([nlist, pad], axis=-1)
-        nsel = nsels[-1]
+        if paddle.is_tensor(nsel):
+            nsel = paddle.to_tensor(nsels[-1], dtype=nsel.dtype)
+        else:
+            nsel = nsels[-1]
+
     # nb x nall x 3
     coord1 = coord.reshape([nb, -1, 3])
     nall = coord1.shape[1]
diff --git a/source/tests/pd/model/test_jit.py b/source/tests/pd/model/test_jit.py
index 20a3f67a71..f4f755d2eb 100644
--- a/source/tests/pd/model/test_jit.py
+++ b/source/tests/pd/model/test_jit.py
@@ -115,7 +115,6 @@ def tearDown(self):
         JITTest.tearDown(self)
 
 
-@unittest.skip("var dtype int32/int64 confused in if block")
 class TestEnergyModelDPA2(unittest.TestCase, JITTest):
     def setUp(self):
         input_json = str(Path(__file__).parent / "water/se_atten.json")
diff --git a/source/tests/pd/test_multitask.py b/source/tests/pd/test_multitask.py
index 4dce6114da..8c7ceb5e90 100644
--- a/source/tests/pd/test_multitask.py
+++ b/source/tests/pd/test_multitask.py
@@ -183,7 +183,6 @@ def tearDown(self):
                 shutil.rmtree(f)
 
 
-@unittest.skip("Paddle do not support MultiTaskSeA.")
 class TestMultiTaskSeA(unittest.TestCase, MultiTaskTrainTest):
     def setUp(self):
         multitask_se_e2_a = deepcopy(multitask_template)