diff --git a/tensorflow/api_template.__init__.py b/tensorflow/api_template.__init__.py
index 2d97ea3ca50..b75ec148ae8 100644
--- a/tensorflow/api_template.__init__.py
+++ b/tensorflow/api_template.__init__.py
@@ -149,4 +149,5 @@ if hasattr(_current_module, 'keras'):
   optimizers = keras.optimizers
   initializers = keras.initializers
 
+compat.v2.compat.v1 = compat.v1
 # pylint: enable=undefined-variable
diff --git a/tensorflow/api_template_v1.__init__.py b/tensorflow/api_template_v1.__init__.py
index 38b635a83e2..4fa92b07051 100644
--- a/tensorflow/api_template_v1.__init__.py
+++ b/tensorflow/api_template_v1.__init__.py
@@ -141,4 +141,6 @@ try:
     vars()['__all__'].remove('compiler')
 except NameError:
   pass
+
+compat.v2.compat.v1 = compat.v1
 # pylint: enable=undefined-variable