diff --git a/flagd/go.mod b/flagd/go.mod
index e0fe6ea88..fcf39ef8a 100644
--- a/flagd/go.mod
+++ b/flagd/go.mod
@@ -8,7 +8,7 @@ require (
 	github.com/mattn/go-colorable v0.1.13
 	github.com/open-feature/flagd/core v0.6.7
 	github.com/open-feature/go-sdk-contrib/providers/flagd v0.1.17
-	github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.0
+	github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.1
 	github.com/spf13/cobra v1.7.0
 	github.com/spf13/viper v1.17.0
 	go.uber.org/zap v1.26.0
diff --git a/flagd/go.sum b/flagd/go.sum
index a299e182e..1981e0381 100644
--- a/flagd/go.sum
+++ b/flagd/go.sum
@@ -694,6 +694,8 @@ github.com/open-feature/go-sdk-contrib/providers/flagd v0.1.17 h1:Ivx0USjWI2iXq2
 github.com/open-feature/go-sdk-contrib/providers/flagd v0.1.17/go.mod h1:ndXCZtJ3xvUK/MQlyHcs8gwW1UQmVMhFGV+OgcUj9TQ=
 github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.0 h1:XstvqjyOAo7s0no2z7R8kN1TrIRWzXtu867h0O+gzkk=
 github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.0/go.mod h1:Hb/nrUmwLdo4j+r+WwdBR0jXFaVrKdbVOHaQgq6ra2A=
+github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.1 h1:oGozIhxePSlE1mvJgWmbYgERHt7BlOFxKCoJgoP2dLI=
+github.com/open-feature/go-sdk-contrib/tests/flagd v1.3.1/go.mod h1:Hb/nrUmwLdo4j+r+WwdBR0jXFaVrKdbVOHaQgq6ra2A=
 github.com/open-feature/open-feature-operator v0.2.36 h1:dzyZh9JSIRvXkfpM9ynYplNk7vjQFLs9sd5aHhF48z4=
 github.com/open-feature/open-feature-operator v0.2.36/go.mod h1:nM7T4oGQukeGmcAFkQm0uwt8WFdDb5hYPjXkm7pHhX4=
 github.com/open-feature/schemas v0.2.8 h1:oA75hJXpOd9SFgmNI2IAxWZkwzQPUDm7Jyyh3q489wM=
diff --git a/flagd/tests/integration/evaluation_test.go b/flagd/tests/integration/evaluation_test.go
index fec27b0cb..99345086a 100644
--- a/flagd/tests/integration/evaluation_test.go
+++ b/flagd/tests/integration/evaluation_test.go
@@ -7,6 +7,7 @@ import (
 	"github.com/cucumber/godog"
 	flagd "github.com/open-feature/go-sdk-contrib/providers/flagd/pkg"
 	"github.com/open-feature/go-sdk-contrib/tests/flagd/pkg/integration"
+	"github.com/open-feature/go-sdk/pkg/openfeature"
 )
 
 func TestEvaluation(t *testing.T) {
@@ -25,8 +26,10 @@ func TestEvaluation(t *testing.T) {
 	}
 
 	testSuite := godog.TestSuite{
-		Name:                name,
-		ScenarioInitializer: integration.InitializeEvaluationScenario(providerOptions...),
+		Name: name,
+		ScenarioInitializer: integration.InitializeEvaluationScenario(func() openfeature.FeatureProvider {
+			return flagd.NewProvider(providerOptions...)
+		}),
 		Options: &godog.Options{
 			Format:   "pretty",
 			Paths:    []string{"../../../spec/specification/assets/gherkin/evaluation.feature"},
diff --git a/flagd/tests/integration/json_evaluator_test.go b/flagd/tests/integration/json_evaluator_test.go
index 22b01fa38..e4c205cb8 100644
--- a/flagd/tests/integration/json_evaluator_test.go
+++ b/flagd/tests/integration/json_evaluator_test.go
@@ -7,6 +7,7 @@ import (
 	"github.com/cucumber/godog"
 	flagd "github.com/open-feature/go-sdk-contrib/providers/flagd/pkg"
 	"github.com/open-feature/go-sdk-contrib/tests/flagd/pkg/integration"
+	"github.com/open-feature/go-sdk/pkg/openfeature"
 )
 
 func TestJsonEvaluator(t *testing.T) {
@@ -20,8 +21,10 @@ func TestJsonEvaluator(t *testing.T) {
 	name := "flagd-json-evaluator.feature"
 
 	testSuite := godog.TestSuite{
-		Name:                name,
-		ScenarioInitializer: integration.InitializeFlagdJsonScenario(providerOptions...),
+		Name: name,
+		ScenarioInitializer: integration.InitializeFlagdJsonScenario(func() openfeature.FeatureProvider {
+			return flagd.NewProvider(providerOptions...)
+		}),
 		Options: &godog.Options{
 			Format:   "pretty",
 			Paths:    []string{"../../../test-harness/gherkin/flagd-json-evaluator.feature"},