diff --git a/python/paddle/v2/fluid/tests/book_distribute/test_understand_sentiment_conv_dist.py b/python/paddle/v2/fluid/tests/book_distribute/test_understand_sentiment_conv_dist.py
index 0f0d84be0a63b5b7004a671cf71b4cbfc84eb5e0..db419e23abcd06ca39011b1bef078b0cafb5100e 100644
--- a/python/paddle/v2/fluid/tests/book_distribute/test_understand_sentiment_conv_dist.py
+++ b/python/paddle/v2/fluid/tests/book_distribute/test_understand_sentiment_conv_dist.py
@@ -1,4 +1,5 @@
 from __future__ import print_function
+import os
 import numpy as np
 import paddle.v2 as paddle
 import paddle.v2.fluid as fluid