diff --git a/examples/partitioned_dag.py b/examples/partitioned_dag.py
index da26392..96ae8e4 100644
--- a/examples/partitioned_dag.py
+++ b/examples/partitioned_dag.py
@@ -4,9 +4,8 @@
 import numpy as np
 import pyarrow as pa
 from dagster import DailyPartitionsDefinition, Definitions, Jitter, RetryPolicy, asset
-from pyiceberg.catalog.sql import SqlCatalog
-
 from dagster_pyiceberg import IcebergPyarrowIOManager, IcebergSqlCatalogConfig
+from pyiceberg.catalog.sql import SqlCatalog
 
 CATALOG_URI = "sqlite:////home/vscode/workspace/.tmp/dag/warehouse/catalog.db"
 CATALOG_WAREHOUSE = "file:///home/vscode/workspace/.tmp/dag/warehouse"
diff --git a/examples/simple_dag.py b/examples/simple_dag.py
index 042f10e..4c2a542 100644
--- a/examples/simple_dag.py
+++ b/examples/simple_dag.py
@@ -3,9 +3,8 @@
 import numpy as np
 import pyarrow as pa
 from dagster import Definitions, asset
-from pyiceberg.catalog.sql import SqlCatalog
-
 from dagster_pyiceberg import IcebergPyarrowIOManager, IcebergSqlCatalogConfig
+from pyiceberg.catalog.sql import SqlCatalog
 
 CATALOG_URI = "sqlite:////home/vscode/workspace/.tmp/dag/warehouse/catalog.db"
 CATALOG_WAREHOUSE = "file:///home/vscode/workspace/.tmp/dag/warehouse"
diff --git a/pyproject.toml b/pyproject.toml
index 7a7165d..7cf8f45 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,6 +57,7 @@ extend_skip = [".notebooks", ".nox", ".venv"]
 [tool.mypy]
 exclude = "^.notebooks/"
 ignore_missing_imports = true
+explicit_package_bases = true
 
 [tool.bandit]
 skips = ['B101']