diff --git a/iam_builder/iam_schema.py b/iam_builder/iam_schema.py
index 07234ce..44da25a 100644
--- a/iam_builder/iam_schema.py
+++ b/iam_builder/iam_schema.py
@@ -4,7 +4,7 @@
 
 try:
     import pkg_resources
-except NameError:
+except ModuleNotFoundError:
     import importlib