diff --git a/ppcls/data/reader.py b/ppcls/data/reader.py
index c1428a36be1010318e03991a962d00a63635a757..cd0bf1350129f196324bf0b5c010c07e3239c45c 100755
--- a/ppcls/data/reader.py
+++ b/ppcls/data/reader.py
@@ -17,7 +17,7 @@ import imghdr
 import os
 import signal
 
-from paddle.fluid.io import multiprocess_reader
+from paddle.reader import multiprocess_reader
 
 from . import imaug
 from .imaug import transform