diff --git a/tensorflow_code/create_pretraining_data.py b/tensorflow_code/create_pretraining_data.py
index b89e5ad2..f10d1290 100644
--- a/tensorflow_code/create_pretraining_data.py
+++ b/tensorflow_code/create_pretraining_data.py
@@ -21,7 +21,7 @@ from __future__ import print_function
 import collections
 import random
 
-import tokenization
+from tensorflow_code import tokenization
 import tensorflow as tf
 
 flags = tf.flags
diff --git a/tensorflow_code/extract_features.py b/tensorflow_code/extract_features.py
index dd6a10b4..65db07d2 100644
--- a/tensorflow_code/extract_features.py
+++ b/tensorflow_code/extract_features.py
@@ -23,8 +23,8 @@ import collections
 import json
 import re
 
-import modeling
-import tokenization
+from tensorflow_code import modeling
+from tensorflow_code import tokenization
 import tensorflow as tf
 
 flags = tf.flags
diff --git a/tensorflow_code/modeling_test.py b/tensorflow_code/modeling_test.py
index bed462f3..f30d7b1d 100644
--- a/tensorflow_code/modeling_test.py
+++ b/tensorflow_code/modeling_test.py
@@ -21,7 +21,7 @@ import json
 import random
 import re
 
-import modeling
+from tensorflow_code import modeling
 import six
 import tensorflow as tf
 
diff --git a/tensorflow_code/optimization_test.py b/tensorflow_code/optimization_test.py
index 266ececd..56ae1f09 100644
--- a/tensorflow_code/optimization_test.py
+++ b/tensorflow_code/optimization_test.py
@@ -16,7 +16,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import optimization
+from tensorflow_code import optimization
 import tensorflow as tf
 
 
diff --git a/tensorflow_code/run_classifier.py b/tensorflow_code/run_classifier.py
index a1ab1002..49e8a191 100644
--- a/tensorflow_code/run_classifier.py
+++ b/tensorflow_code/run_classifier.py
@@ -20,9 +20,9 @@ from __future__ import print_function
 
 import csv
 import os
-import modeling
-import optimization
-import tokenization
+from tensorflow_code import modeling
+from tensorflow_code import optimization
+from tensorflow_code import tokenization
 import tensorflow as tf
 
 flags = tf.flags
diff --git a/tensorflow_code/run_pretraining.py b/tensorflow_code/run_pretraining.py
index 4636f205..f358366e 100644
--- a/tensorflow_code/run_pretraining.py
+++ b/tensorflow_code/run_pretraining.py
@@ -19,8 +19,8 @@ from __future__ import division
 from __future__ import print_function
 
 import os
-import modeling
-import optimization
+from tensorflow_code import modeling
+from tensorflow_code import optimization
 import tensorflow as tf
 
 flags = tf.flags
diff --git a/tensorflow_code/run_squad.py b/tensorflow_code/run_squad.py
index 642a1878..54548a02 100644
--- a/tensorflow_code/run_squad.py
+++ b/tensorflow_code/run_squad.py
@@ -22,9 +22,9 @@ import collections
 import json
 import math
 import os
-import modeling
-import optimization
-import tokenization
+from tensorflow_code import modeling
+from tensorflow_code import optimization
+from tensorflow_code import tokenization
 import six
 import tensorflow as tf
 
diff --git a/tensorflow_code/tokenization_test.py b/tensorflow_code/tokenization_test.py
index 9a6e7693..90a1b988 100644
--- a/tensorflow_code/tokenization_test.py
+++ b/tensorflow_code/tokenization_test.py
@@ -19,7 +19,7 @@ from __future__ import print_function
 import os
 import tempfile
 
-import tokenization
+from tensorflow_code import tokenization
 import tensorflow as tf