diff --git a/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py b/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py
index b2bf6e546e724fbe61e78ce792c951a4f8aeb40a..bca91c536ba32b05138f2860c13fdd1899a2e011 100644
--- a/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_lookup_sparse_table_fuse_ops.py
@@ -94,8 +94,10 @@ class TestLookupTableFuseOp(unittest.TestCase):
 
         training_program = fluid.Program()
 
-        scope.var('Beta1Pow').get_tensor().set([0], place)
-        scope.var('Beta2Pow').get_tensor().set([0], place)
+        scope.var('Beta1Pow').get_tensor().set(
+            np.array([0]).astype("float32"), place)
+        scope.var('Beta2Pow').get_tensor().set(
+            np.array([0]).astype("float32"), place)
 
         rows = [0, 1, 2, 3, 4, 5, 6]
         row_numel = 8