diff --git a/tensorflow/lite/schema/upgrade_schema.py b/tensorflow/lite/schema/upgrade_schema.py
index dfb38ec1d93658694fc4e57dc2c72c27fd1abe20..2aa5c6e0d6884c94e1fb9dbbdfbe0691cd6fadd3 100644
--- a/tensorflow/lite/schema/upgrade_schema.py
+++ b/tensorflow/lite/schema/upgrade_schema.py
@@ -348,4 +348,4 @@ def main(argv):
 
 if __name__ == "__main__":
   FLAGS, unparsed = parser.parse_known_args()
-  tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
+  tf.compat.v1.app.run(main=main, argv=[sys.argv[0]] + unparsed)
diff --git a/tensorflow/lite/testing/generate_examples.py b/tensorflow/lite/testing/generate_examples.py
index df7ce293b91ed40a4b970b6f26bc29747404a6d4..5d8662d79395e8a34ee9cb1918713c0bf34a92f4 100644
--- a/tensorflow/lite/testing/generate_examples.py
+++ b/tensorflow/lite/testing/generate_examples.py
@@ -105,4 +105,4 @@ if __name__ == "__main__":
   if unparsed:
     print("Usage: %s <path out> <zip file to generate>")
   else:
-    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
+    tf.compat.v1.app.run(main=main, argv=[sys.argv[0]] + unparsed)
diff --git a/tensorflow/lite/tutorials/mnist_tflite.py b/tensorflow/lite/tutorials/mnist_tflite.py
index 6434277f096b494925a7db7e3ac2e9190af19366..60df2664dcdaf0fe2fe03e69dd87f9105a40e00b 100644
--- a/tensorflow/lite/tutorials/mnist_tflite.py
+++ b/tensorflow/lite/tutorials/mnist_tflite.py
@@ -84,4 +84,4 @@ def main(_):
 
 if __name__ == '__main__':
   tf.logging.set_verbosity(tf.logging.INFO)
-  tf.app.run(main)
+  tf.compat.v1.app.run(main)