diff --git a/tests/test_nn_Module_type.py b/tests/test_nn_Module_type.py
index 742212dba..ecb0c400e 100644
--- a/tests/test_nn_Module_type.py
+++ b/tests/test_nn_Module_type.py
@@ -26,7 +26,7 @@ def test_case_1():
         x = torch.tensor([1., 2., 3.])
         module1 = torch.nn.Module()
         module1.register_buffer('buffer', x)
-        module1.type(torch.float32)
+        module1.type_unsupport(torch.float32)
         result = module1.buffer
         """
     )
@@ -45,7 +45,7 @@ def test_case_2():
         x = torch.tensor([1., 2., 3.])
         module1 = torch.nn.Module()
         module1.register_buffer('buffer', x)
-        module1.type(dst_type=torch.float32)
+        module1.type_unsupport(dst_type=torch.float32)
         result = module1.buffer
         """
     )