diff --git a/server/visualdl/lib.py b/server/visualdl/lib.py
index ac138f6a2719ff06d139d907c92306579feced68..3764b58fb1bfd05d3dfe7eaf2849070f41de21d9 100644
--- a/server/visualdl/lib.py
+++ b/server/visualdl/lib.py
@@ -89,7 +89,7 @@ def get_image_tag_steps(storage, mode, tag):
         record = image.record(step_index, sample_index)
         shape = record.shape()
         # TODO(ChunweiYan) remove this trick, some shape will be empty
-        # if not shape: continue
+        if not shape: continue
         # assert shape, "%s,%s" % (mode, tag)
         query = urllib.urlencode({
             'sample': 0,