diff --git a/pyproject.toml b/pyproject.toml
index 1cc23479..794077d3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "reasoning_gym"
-version = "0.1.4"
+version = "0.1.5"
 authors = [
   { name = "Open-Thought community", email = "andreas.koepf@xamla.com" },
 ]
@@ -31,20 +31,20 @@ license = "Apache-2.0"
 license-files = ["LICENSE*"]
 
 [project.optional-dependencies]
-test = [
-    "pytest>=7.0.0",
-    "pytest-cov>=4.0.0",
-]
+test = ["pytest>=7.0.0", "pytest-cov>=4.0.0"]
 
 [project.urls]
 "Homepage" = "https://github.com/open-thought/reasoning-gym"
 "Bug Tracker" = "https://github.com/open-thought/reasoning-gym/issues"
 
-[tool.hatch.build.targets.wheel]
-packages = ["reasoning_gym"]
 
 [tool.hatch.build]
-include = ["reasoning_gym/**/*.txt"]
+packages = ["reasoning_gym"]
+include = [
+  "reasoning_gym/**/*.py",
+  "reasoning_gym/**/*.txt",
+  "reasoning_gym/**/levels/*",
+]
 
 [tool.black]
 line-length = 120
@@ -58,6 +58,4 @@ line_length = 120
 
 [tool.pytest.ini_options]
 addopts = "-ra -q"
-testpaths = [
-    "tests",
-]
+testpaths = ["tests"]
diff --git a/reasoning_gym/__init__.py b/reasoning_gym/__init__.py
index 6f18d2e7..ecca7f3f 100644
--- a/reasoning_gym/__init__.py
+++ b/reasoning_gym/__init__.py
@@ -5,7 +5,7 @@ Reasoning Gym - A library of procedural dataset generators for training reasonin
 from . import algebra, algorithmic, arithmetic, code, cognition, data, games, geometry, graphs, logic
 from .factory import create_dataset, register_dataset
 
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 __all__ = [
     "algebra",
     "algorithmic",