diff --git a/pyproject.toml b/pyproject.toml
index fce9d11..40c020d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -80,3 +80,7 @@ extend-select = [
     "RUF", # Ruff-specific
     "UP", # pyupgrade
 ]
+
+[tool.setuptools]
+packages = ["pylibbpf"]
+package-data = {"pylibbpf" = ["*.py", "py.typed"]}
diff --git a/setup.py b/setup.py
index ae74c74..02e0658 100644
--- a/setup.py
+++ b/setup.py
@@ -116,5 +116,4 @@ if readme_path.exists():
 setup(
     ext_modules=[CMakeExtension("pylibbpf.pylibbpf")],
     cmdclass={"build_ext": CMakeBuild},
-    zip_safe=False,
 )