diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 0684ecf..b090ae5 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -27,8 +27,6 @@ repos:
       - id: check-yaml
       - id: check-merge-conflict
       - id: detect-private-key
-      - id: no-commit-to-branch
-        args: ["--branch=main"]
   - repo: https://github.com/codespell-project/codespell
     rev: v2.4.1
     hooks:
diff --git a/src/arrayloader_benchmarks/__init__.py b/arrayloader_benchmarks/__init__.py
similarity index 100%
rename from src/arrayloader_benchmarks/__init__.py
rename to arrayloader_benchmarks/__init__.py
diff --git a/src/arrayloader_benchmarks/compute_spec.py b/arrayloader_benchmarks/compute_spec.py
similarity index 100%
rename from src/arrayloader_benchmarks/compute_spec.py
rename to arrayloader_benchmarks/compute_spec.py
diff --git a/hatch.toml b/hatch.toml
deleted file mode 100644
index 12c63b0..0000000
--- a/hatch.toml
+++ /dev/null
@@ -1,13 +0,0 @@
-[envs.default]
-installer = "uv"
-features = [ "test" ]
-
-[envs.arrayloaders]
-features = [ "arrayloaders" ]
-scripts.generate_data = "echo TODO"
-scripts.benchmark = "echo {args}"
-
-[envs.hatch-test]
-default-args = [  ]
-features = [ "test" ]
-extra-dependencies = [ "ipykernel" ]
diff --git a/pyproject.toml b/pyproject.toml
index f3eedf0..3fddbc9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -44,10 +44,10 @@ test = ["pytest", "pytest-xdist"]
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/arrayloader_benchmarks"]
+packages = ["arrayloader_benchmarks"]
 
 [tool.ruff]
-src = ["src"]
+src = ["."]
 
 [tool.ruff.format]
 docstring-code-format = true
diff --git a/scripts/run_loading_benchmark_on_collection.py b/scripts/run_loading_benchmark_on_collection.py
index 5efa739..749d677 100644
--- a/scripts/run_loading_benchmark_on_collection.py
+++ b/scripts/run_loading_benchmark_on_collection.py
@@ -262,7 +262,11 @@ def run(
         df = pd.concat([df, pd.DataFrame([new_result])], ignore_index=True)
     except ln.Artifact.DoesNotExist:
         df = pd.DataFrame([new_result])
-    ln.Artifact.from_dataframe(df, key=results_key).save()
+    if ln.setup.settings.instance.slug == "laminlabs/arrayloader-benchmarks":
+        storage = ln.Storage.get(root="s3://lamin-us-west-2/wXDsTYYd")
+    else:
+        storage = None
+    ln.Artifact.from_dataframe(df, key=results_key, storage=storage).save()
 
 
 if __name__ == "__main__":