diff --git a/pyproject.toml b/pyproject.toml
index a2466a8..58fe563 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,13 +10,12 @@ authors = [
 ]
 description = "A python package for deep learning based image to image transformation"
 readme = "README.md"
-license = { text="MIT license" }
+license = "MIT"
 requires-python = ">=3.10"
 keywords = ["deep learning", "microscopy image analysis", "biomedical image analysis"]
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Developers",
-    "License :: OSI Approved :: MIT License",
     "Natural Language :: English",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
@@ -126,6 +125,9 @@ dynamic = ["optional-dependencies"]
 include-package-data = true
 zip-safe = false
 
+[tool.setuptools.packages.find]
+include = ["mmv_im2im"]
+
 [project.scripts]
 run_im2im = "mmv_im2im.bin.run_im2im:main"