diff --git a/packages/squiggle-lang/scripts/bench-sampleset-to-pointset.mjs b/packages/squiggle-lang/scripts/bench-sampleset-to-pointset.mjs
index 06400f9e..7fc87dea 100755
--- a/packages/squiggle-lang/scripts/bench-sampleset-to-pointset.mjs
+++ b/packages/squiggle-lang/scripts/bench-sampleset-to-pointset.mjs
@@ -4,19 +4,31 @@ import { measure } from "./lib.mjs";
 
 const maxP = 3;
 
+const sampleCount = process.env.SAMPLE_COUNT;
+
 for (let p = 0; p <= maxP; p++) {
   const size = Math.pow(10, p);
   const project = SqProject.create();
+  if (sampleCount) {
+    project.setEnvironment({
+      sampleCount: Number(sampleCount),
+      xyPointLength: Number(sampleCount),
+    });
+  }
   project.setSource(
     "main",
     `
-    List.upTo(1, ${size}) -> map({|x|
-        normal(x,2) -> SampleSet.fromDist -> PointSet.fromDist
-    })->List.last
+    List.upTo(1, ${size}) -> map(
+      { |x| normal(x,2) -> SampleSet.fromDist -> PointSet.fromDist }
+    )->List.last
     `
   );
   const time = measure(() => {
     project.run("main");
   });
+  const result = project.getResult("main");
+  if (result.tag != "Ok") {
+    throw new Error("Code failed: " + result.value.toString());
+  }
   console.log(`1e${p}`, "\t", time);
 }