diff --git a/demo/demo3/run.sh b/demo/demo3/run.sh
index 87eccb9a02237114a1c6e721b16893426e7d66ae..37e95926d7abe9862152b860c60a54759d450942 100755
--- a/demo/demo3/run.sh
+++ b/demo/demo3/run.sh
@@ -1,4 +1,4 @@
-export CUDA_VISIBLE_DEVICES=4
+export CUDA_VISIBLE_DEVICES=3
 
 python run.py
 
diff --git a/paddlepalm/reader/cls.py b/paddlepalm/reader/cls.py
index b1ae96be081f29cdb0febc195f94521cbf207957..9af96b3746e7fef7794fae7b96ae912e46b2dffe 100644
--- a/paddlepalm/reader/cls.py
+++ b/paddlepalm/reader/cls.py
@@ -71,7 +71,7 @@ class ClassifyReader(BaseReader):
         self._batch_size = batch_size
         self._num_epochs = num_epochs
         self._data_generator = self._reader.data_generator( \
-            input_file, batch_size, num_epochs if phase == 'train' else 1, \
+            input_file, batch_size, num_epochs if self._phase == 'train' else 1, \
             shuffle=shuffle_train if self._phase == 'train' else False, \
             phase=self._phase)