diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_one_hot.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_one_hot.py
index 53574a3fd27dc..30446265a431d 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_one_hot.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_one_hot.py
@@ -59,7 +59,7 @@ def generate_depth(dims, batch):
                         },
                         "op_outputs": {"Out": ["output_data"]},
                         "op_attrs": dics[0],
-                        "outputs_dtype": {"output_data": np.int64},
+                        "outputs_dtype": {"output_data": np.int_},
                     },
                 ]
                 ops = self.generate_op_config(ops_config)