diff --git a/appendix-A/02_installing-python-libraries/python_environment_check.py b/appendix-A/02_installing-python-libraries/python_environment_check.py
index 34b25a5..530812e 100644
--- a/appendix-A/02_installing-python-libraries/python_environment_check.py
+++ b/appendix-A/02_installing-python-libraries/python_environment_check.py
@@ -3,7 +3,8 @@
 #   - https://www.manning.com/books/build-a-large-language-model-from-scratch
 # Code: https://github.com/rasbt/LLMs-from-scratch
 
-import importlib
+from importlib.metadata import PackageNotFoundError, import_module
+import importlib.metadata
 from os.path import dirname, join, realpath
 from packaging.version import parse as version_parse
 import platform
@@ -20,7 +21,7 @@ def get_packages(pkgs):
     versions = []
     for p in pkgs:
         try:
-            imported = importlib.import_module(p)
+            imported = import_module(p)
             try:
                 version = (getattr(imported, '__version__', None) or
                            getattr(imported, 'version', None) or
@@ -29,7 +30,7 @@ def get_packages(pkgs):
                     # If common attributes don't exist, use importlib.metadata
                     version = importlib.metadata.version(p)
                 versions.append(version)
-            except importlib.metadata.PackageNotFoundError:
+            except PackageNotFoundError:
                 # Handle case where package is not installed
                 versions.append('0.0')
         except ImportError:
diff --git a/ch02/02_bonus_bytepair-encoder/compare-bpe-tiktoken.ipynb b/ch02/02_bonus_bytepair-encoder/compare-bpe-tiktoken.ipynb
index 133c063..273404a 100644
--- a/ch02/02_bonus_bytepair-encoder/compare-bpe-tiktoken.ipynb
+++ b/ch02/02_bonus_bytepair-encoder/compare-bpe-tiktoken.ipynb
@@ -63,9 +63,9 @@
     }
    ],
    "source": [
-    "import importlib.metadata\n",
+    "from importlib.metadata import version\n",
     "\n",
-    "print(\"tiktoken version:\", importlib.metadata.version(\"tiktoken\"))"
+    "print(\"tiktoken version:\", version(\"tiktoken\"))"
    ]
   },
   {