fix(): typo

knownsec · Sep 7, 2021 · 5339ebd · 5339ebd
1 parent 3a800df
commit 5339ebd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pocsuite3/lib/core/register.py b/pocsuite3/lib/core/register.py
@@ -56,7 +56,10 @@ def check_requires(data):
                     if not import_name:
                         import_name = install_name
                     __import__(import_name)
-                    ver = pkg_resources.get_distribution(import_name).version
+                    try:
+                        ver = pkg_resources.get_distribution(install_name).version
+                    except Exception:
+                        ver = 'unknown'
                     logger.info(f'{install_name}=={ver} has been installed')
             except ImportError:
                 err_msg = f'{install_name} not found, try install with "python -m pip install {install_name}"'