diff --git a/tokenization_test.py b/tokenization_test.py
index 0afaedd2e..6e566af1b 100644
--- a/tokenization_test.py
+++ b/tokenization_test.py
@@ -17,12 +17,14 @@
 from __future__ import print_function
 
 import os
+import sys
+import os.path
 import tempfile
 import tokenization
 import six
 import tensorflow as tf
 
-
+# create class TokenizationTest(tf.test.TestCase):
 class TokenizationTest(tf.test.TestCase):
 
   def test_full_tokenizer(self):