diff --git a/hapi/metrics.py b/hapi/metrics.py
index 1031e48bc36fb4b0b78e8c192553fa17022aa348..5f79b8f11d6427156e53d99581e21eea8a2aad94 100644
--- a/hapi/metrics.py
+++ b/hapi/metrics.py
@@ -56,9 +56,9 @@ class Metric(object):
                        |                      ||
                  outputs labels               ||
                        |                      ||    tensor data
-            {Metric.add_metric_op}            ||
+             {Metric.add_metric_op}           ||
                        |                      ||
-                  metric states               ||
+              metric states(tensor)           ||
                        |                      ||
                 {fetch as numpy}              || ------------------
                        |                      ||