diff --git a/models/rank/dcn/data/get_slot_data.py b/models/rank/dcn/data/get_slot_data.py
index 96d4448214d6a87092495326646a279657079f45..b1adb02817f1abbabfecd328e85aea3ee9136533 100755
--- a/models/rank/dcn/data/get_slot_data.py
+++ b/models/rank/dcn/data/get_slot_data.py
@@ -59,7 +59,7 @@ class TrainReader(dg.MultiSlotDataGenerator):
         self.cat_feat_idx_dict_list = [{} for _ in range(26)]
 
         # TODO: set vocabulary dictionary
-        vocab_dir = "./vocab/"
+        vocab_dir = "./sample_data/vocab/"
         for i in range(26):
             lookup_idx = 1  # remain 0 for default value
             for line in open(
diff --git a/models/rank/wide_deep/data/get_slot_data.py b/models/rank/wide_deep/data/get_slot_data.py
index 831d05665b01649f22a3270ec949ebda2941928d..ec37f9af9ebc1120294f965cc5845ce7bb3feaf0 100755
--- a/models/rank/wide_deep/data/get_slot_data.py
+++ b/models/rank/wide_deep/data/get_slot_data.py
@@ -11,7 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import yaml
+import yaml, os
+
 from paddlerec.core.reader import Reader
 from paddlerec.core.utils import envs
 try:
diff --git a/models/rank/xdeepfm/data/get_slot_data.py b/models/rank/xdeepfm/data/get_slot_data.py
index 4426e9647c080dce5debdcdbc3e039ac69a69935..0ace10b4fada14d1c6b91dfbc4c80bf91737c28d 100755
--- a/models/rank/xdeepfm/data/get_slot_data.py
+++ b/models/rank/xdeepfm/data/get_slot_data.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import yaml
+import yaml, os
 from paddlerec.core.reader import Reader
 from paddlerec.core.utils import envs
 try: