diff --git a/server/visualdl/lib_test.py b/server/visualdl/lib_test.py
index 117cc2d89148b11d2cfc51aeb46f133bc8dc3a02..88f3cd68f30c2de9eec8e024c1f58594d66ee8f1 100644
--- a/server/visualdl/lib_test.py
+++ b/server/visualdl/lib_test.py
@@ -24,7 +24,7 @@ class LibTest(unittest.TestCase):
         add_image(writer, "train", "layer/image1", 7, 10, 1, shape=[30,30,2])
         add_image(writer, "test", "layer/image1", 7, 10, 1, shape=[30,30,2])
 
-        self.reader = storage.StorageReader(dir)
+        self.reader = storage.LogReader(dir)
 
     def test_modes(self):
         modes = lib.get_modes(self.reader)