diff --git a/fluid/text_classification/infer.py b/fluid/text_classification/infer.py
index d2a0363d786866a92195dba8b490287b3ca9bc9d..923dd98dc27f6a4c5e369f30d9aca49e63aea7a5 100644
--- a/fluid/text_classification/infer.py
+++ b/fluid/text_classification/infer.py
@@ -4,8 +4,8 @@ import unittest
 import contextlib
 import numpy as np
 
+import paddle
 import paddle.fluid as fluid
-import paddle.v2 as paddle
 
 import utils