diff --git a/python/paddle/v2/fluid/tests/test_registry.py b/python/paddle/v2/fluid/tests/test_registry.py
index 4fddb91268ee2343adfb1d70e494715611c9c87e..44e50ca55ac609ed2e0a145ff12248fa18479668 100644
--- a/python/paddle/v2/fluid/tests/test_registry.py
+++ b/python/paddle/v2/fluid/tests/test_registry.py
@@ -28,4 +28,4 @@ class TestRegistry(unittest.TestCase):
         exe = fluid.Executor(place)
         X = np.random.random((10, 10)).astype("float32")
         mean_out = exe.run(feed={"X": X}, fetch_list=[output])
-        self.assertAlmostEqual(np.mean(X), mean_out)
+        self.assertAlmostEqual(np.mean(X), mean_out[0])