diff --git a/tensorflow/python/keras/__init__.py b/tensorflow/python/keras/__init__.py
index 6f79c219867..47f207329d7 100644
--- a/tensorflow/python/keras/__init__.py
+++ b/tensorflow/python/keras/__init__.py
@@ -32,13 +32,6 @@ from tensorflow.python.keras.engine.training import Model
 
 from tensorflow.python.util.tf_export import keras_export
 
-if tf2.enabled():
-  __version__ = '2.3.0-tf'
-else:
-  __version__ = '2.2.4-tf'
+__version__ = '2.4.0'
 
 keras_export('keras.__version__').export_constant(__name__, '__version__')
-
-del absolute_import
-del division
-del print_function