diff --git a/pyproject.toml b/pyproject.toml
index dc811127..b7114e94 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -95,9 +95,17 @@ requires = [
 ]
 build-backend = "setuptools.build_meta"
 
+[tool.setuptools.packages.find]
+include = ["econml", "econml.*"]
+exclude = ["econml.tests"]
+
+[tool.setuptools.package-data]
+# include all CSV files as data
+"*" = ["*.csv", "*.jbl"]
+
 [tool.pytest.ini_options]
 testpaths = ["econml/tests"]
-addopts = "--junitxml=junit/test-results.xml -n auto --strict-markers --cov-config=pyproject.toml --cov"
+addopts = "--junitxml=junit/test-results.xml -n auto --strict-markers --cov-config=pyproject.toml --cov --import-mode=importlib"
 markers = [    
     "slow",
     "notebook",
diff --git a/setup.cfg b/setup.cfg
index 378a711a..24c9bfec 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -6,17 +6,4 @@ max-line-length=119
 ; Use numpy style
 convention=numpy
 
-[options]
-packages = find_namespace:
-    
-[options.packages.find]
-include =
-    econml
-    econml.*
-exclude = 
-    econml.tests
 
-[options.package_data]
-; include all CSV files as data
-* = *.csv
-    *.jbl