diff --git a/flow/e2e/test_utils.go b/flow/e2e/test_utils.go
index 26db73f765..659b5249ac 100644
--- a/flow/e2e/test_utils.go
+++ b/flow/e2e/test_utils.go
@@ -298,7 +298,7 @@ func CreateQRepWorkflowConfig(
 }
 
 func RunQrepFlowWorkflow(env *testsuite.TestWorkflowEnvironment, config *protos.QRepConfig) {
-	state := peerflow.NewQRepFlowStateForTesting()
+	state := peerflow.NewQRepFlowState()
 	time.Sleep(5 * time.Second)
 	env.ExecuteWorkflow(peerflow.QRepFlowWorkflow, config, state)
 }