diff --git a/python/paddle/fluid/tests/unittests/test_one_hot_new_op.py b/python/paddle/fluid/tests/unittests/test_one_hot_new_op.py
index 339f689998f817054611bd85b11945b61d1f649b..181c0d4b0e76c67a105a3221058fcf188ac34140 100644
--- a/python/paddle/fluid/tests/unittests/test_one_hot_new_op.py
+++ b/python/paddle/fluid/tests/unittests/test_one_hot_new_op.py
@@ -45,7 +45,7 @@ class TestOneHotOp(OpTest):
         self.outputs = {'Out': (out, x_lod)}
 
     def test_check_output(self):
-        self.check_output(check_dygraph=False)
+        self.check_output(check_dygraph=True)
 
 
 class TestOneHotOp_attr(OpTest):
@@ -68,7 +68,7 @@ class TestOneHotOp_attr(OpTest):
         self.outputs = {'Out': (out, x_lod)}
 
     def test_check_output(self):
-        self.check_output(check_dygraph=False)
+        self.check_output(check_dygraph=True)
 
 
 class TestOneHotOp_default_dtype(OpTest):
@@ -91,7 +91,7 @@ class TestOneHotOp_default_dtype(OpTest):
         self.outputs = {'Out': (out, x_lod)}
 
     def test_check_output(self):
-        self.check_output(check_dygraph=False)
+        self.check_output(check_dygraph=True)
 
 
 class TestOneHotOp_default_dtype_attr(OpTest):
@@ -114,7 +114,7 @@ class TestOneHotOp_default_dtype_attr(OpTest):
         self.outputs = {'Out': (out, x_lod)}
 
     def test_check_output(self):
-        self.check_output(check_dygraph=False)
+        self.check_output(check_dygraph=True)
 
 
 class TestOneHotOp_exception(unittest.TestCase):