diff --git a/packages/core/pattern.mjs b/packages/core/pattern.mjs
index d0be3f599..72f6371b3 100644
--- a/packages/core/pattern.mjs
+++ b/packages/core/pattern.mjs
@@ -98,10 +98,7 @@ export class Pattern {
 
   // runs func on query state
   withState(func) {
-    return this.withHaps((haps, state) => {
-      func(state);
-      return haps;
-    });
+    return new Pattern((state) => this.query(func(state)));
   }
 
   /**