diff --git a/doc/conf.py b/doc/conf.py
index 330ebaed7b..9872774159 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -13,7 +13,7 @@
 """Configuration for generating readthedocs docstrings."""
 from __future__ import absolute_import
 
-import pkg_resources
+from importlib.metadata import version as _get_distribution_version
 from datetime import datetime
 import sys
 import os
@@ -22,7 +22,7 @@
 from doc_utils.jumpstart_doc_utils import create_jumpstart_model_table  # noqa: E402
 
 project = "sagemaker"
-version = pkg_resources.require(project)[0].version
+version = _get_distribution_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/pyproject.toml b/pyproject.toml
index 0621d9a643..12040ebc5d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,6 +47,7 @@ dependencies = [
   "platformdirs",
   "protobuf>=3.12,<7.0",
   "psutil",
+  "pytz",
   "PyYAML>=6.0.1",
   "requests",
   "sagemaker-core>=1.0.71,<2.0.0",