diff --git a/tests/models/dpa2.pth b/tests/models/dpa2.pt
similarity index 50%
rename from tests/models/dpa2.pth
rename to tests/models/dpa2.pt
index 0559d30c..ac0b0237 100644
Binary files a/tests/models/dpa2.pth and b/tests/models/dpa2.pt differ
diff --git a/tests/test_descriptor_dpa2.py b/tests/test_descriptor_dpa2.py
index 37228808..bca10248 100644
--- a/tests/test_descriptor_dpa2.py
+++ b/tests/test_descriptor_dpa2.py
@@ -37,7 +37,7 @@ def setUp(self):
     ]).to(env.DEVICE)
     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.pth"
+    self.file_model_param = Path(CUR_DIR)/"models"/"dpa2.pt"
     self.file_type_embed = Path(CUR_DIR)/"models"/"dpa2_tebd.pth"
 
   def test_descriptor_hyb(self):