diff --git a/setup.cfg b/setup.cfg
index 033d2fccb..1cb92f78a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -98,6 +98,10 @@ ignore = E203, E501, W503
 [mypy]
 check_untyped_defs = True
 plugins = pydantic.mypy
+exclude = ^src/gt4sd/frameworks/gflownet/envs
+
+[mypy-gt4sd.frameworks.gflownet.envs.*]
+follow_imports = skip
 
 [mypy-pytest.*]
 ignore_missing_imports = True
@@ -105,6 +109,9 @@ ignore_missing_imports = True
 [mypy-rdkit.*]
 ignore_missing_imports = True
 
+[mypy-rdkit-stubs.*]
+ignore_errors = True
+
 [mypy-setuptools.*]
 ignore_missing_imports = True
 
@@ -141,7 +148,6 @@ ignore_missing_imports = True
 [mypy-transformers.*]
 ignore_missing_imports = True
 
-# to avoid mypy from crashing (https://github.com/python/mypy/issues/11045)
 [mypy-transformers.trainer]
 check_untyped_defs = False
 
@@ -284,7 +290,4 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-xgboost.*]
-ignore_missing_imports = True
-
-[mypy-rdkit-stubs.*]
-ignore_errors = True
\ No newline at end of file
+ignore_missing_imports = True
\ No newline at end of file