diff --git a/python/ngen_cal/src/ngen/cal/__main__.py b/python/ngen_cal/src/ngen/cal/__main__.py
index 73e2ed10..336bf259 100644
--- a/python/ngen_cal/src/ngen/cal/__main__.py
+++ b/python/ngen_cal/src/ngen/cal/__main__.py
@@ -7,9 +7,7 @@
 from ngen.cal.strategy import Algorithm
 from ngen.cal.agent import Agent
 
-from typing import TYPE_CHECKING
-if TYPE_CHECKING:
-    from typing import Mapping, Any
+from typing import Mapping, Any
 
 def main(general: General, model_conf: Mapping[str, Any]):
     #seed the random number generators if requested
diff --git a/python/ngen_cal/src/ngen/cal/_version.py b/python/ngen_cal/src/ngen/cal/_version.py
index 7fd229a3..fc79d63d 100644
--- a/python/ngen_cal/src/ngen/cal/_version.py
+++ b/python/ngen_cal/src/ngen/cal/_version.py
@@ -1 +1 @@
-__version__ = '0.2.0'
+__version__ = '0.2.1'