diff --git a/pyproject.toml b/pyproject.toml
index f1af73d6..f5ff80d1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -144,15 +144,11 @@ exclude_dirs = ["*/tests/*", "*/testing/*"]
 zip-safe = true
 
 [tool.setuptools.package-data]
-sktime = [
+skbase = [
     "*.csv",
     "*.csv.gz",
-    "*.arff",
-    "*.arff.gz",
     "*.txt",
-    "*.ts",
-    "*.tsv",
 ]
 
 [tool.setuptools.packages.find]
-exclude = ["tests", "tests.*"]
+include = ["skbase", "skbase.*"]