diff --git a/doc/conf.py b/doc/conf.py
index 2889cfa..230645b 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -13,9 +13,9 @@
 from __future__ import absolute_import
 
 import os
-import pkg_resources
 import sys
 from datetime import datetime
+from importlib.metadata import version as importlib_version
 from unittest.mock import MagicMock
 
 class Mock(MagicMock):
@@ -50,7 +50,7 @@ def __getattr__(cls, name):
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 
 project = u"stepfunctions"
-version = pkg_resources.require(project)[0].version
+version = importlib_version(project)
 
 # Add any Sphinx extension module names here, as strings. They can be extensions
 # coming with Sphinx (named 'sphinx.ext.*') or your custom ones.
diff --git a/src/stepfunctions/__init__.py b/src/stepfunctions/__init__.py
index a1711c3..f4a7ba1 100644
--- a/src/stepfunctions/__init__.py
+++ b/src/stepfunctions/__init__.py
@@ -13,10 +13,11 @@
 from __future__ import absolute_import
 
 import logging
-import pkg_resources
 import sys
 
-__version__ = pkg_resources.require("stepfunctions")[0].version
+from importlib.metadata import version as importlib_version
+
+__version__ = importlib_version("stepfunctions")
 __useragent__ = "aws-step-functions-data-science-sdk-python"
 
 # disable logging.warning() from import packages