diff --git a/tensorflow_hub/__init__.py b/tensorflow_hub/__init__.py
index 1bccbe29..973010bb 100644
--- a/tensorflow_hub/__init__.py
+++ b/tensorflow_hub/__init__.py
@@ -58,7 +58,7 @@ def _ensure_tf_install():
         "https://tensorflow.org/hub/installation.\n\n")
     raise
 
-  from pkg_resources import parse_version
+  from packaging.version import parse as parse_version
 
   #
   # Update this whenever we need to depend on a newer TensorFlow release.