diff --git a/promql/promqltest/test.go b/promql/promqltest/test.go
index e078bcb60b..09f9386f5b 100644
--- a/promql/promqltest/test.go
+++ b/promql/promqltest/test.go
@@ -185,6 +185,11 @@ func newTest(t testutil.T, input string) (*test, error) {
 	return test, err
 }
 
+// SetStorage sets test's storage
+func (t *test) SetStorage(s storage.Storage) {
+	t.storage = s
+}
+
 //go:embed testdata
 var testsFs embed.FS