diff --git a/source/tests/pt/models/dpa2_tebd.pth b/source/tests/pt/models/dpa2_tebd.pth
new file mode 100644
index 0000000000..3d4fc5511c
Binary files /dev/null and b/source/tests/pt/models/dpa2_tebd.pth differ
diff --git a/source/tests/pt/test_descriptor_dpa2.py b/source/tests/pt/test_descriptor_dpa2.py
index 093f53bc32..22371f612d 100644
--- a/source/tests/pt/test_descriptor_dpa2.py
+++ b/source/tests/pt/test_descriptor_dpa2.py
@@ -112,7 +112,7 @@ def setUp(self):
         with open(Path(CUR_DIR) / "models" / "dpa2_hyb.json") as fp:
             self.model_json = json.load(fp)
         self.file_model_param = Path(CUR_DIR) / "models" / "dpa2.pt"
-        self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pt"
+        self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pth"
 
     def test_descriptor_hyb(self):
         # torch.manual_seed(0)