diff --git a/go.mod b/go.mod
index 0b49389b7..78012d4aa 100644
--- a/go.mod
+++ b/go.mod
@@ -10,6 +10,7 @@ require (
 	github.com/stretchr/testify v1.4.0
 	github.com/twmb/murmur3 v1.0.0
 	golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e
+	github.com/cucumber/messages-go/v10 v10.0.1
 )
 
 // Work around issue wtih git.apache.org/thrift.git
diff --git a/tests/integration/main_test.go b/tests/integration/main_test.go
index cb8d7d03e..54b063ae1 100644
--- a/tests/integration/main_test.go
+++ b/tests/integration/main_test.go
@@ -21,8 +21,8 @@ import (
 	"os"
 	"testing"
 
-	"github.com/DATA-DOG/godog"
-	"github.com/DATA-DOG/godog/colors"
+	"github.com/cucumber/godog"
+	"github.com/cucumber/godog/colors"
 	"github.com/optimizely/go-sdk/tests/integration/support"
 )
 
diff --git a/tests/integration/support/steps.go b/tests/integration/support/steps.go
index 2606623d7..3755395ac 100644
--- a/tests/integration/support/steps.go
+++ b/tests/integration/support/steps.go
@@ -24,7 +24,8 @@ import (
 
 	"github.com/optimizely/go-sdk/pkg/decision"
 
-	"github.com/DATA-DOG/godog/gherkin"
+	"github.com/cucumber/gherkin-go"
+	"github.com/cucumber/messages-go"
 	"github.com/google/uuid"
 	"github.com/optimizely/go-sdk/pkg/entities"
 	"github.com/optimizely/go-sdk/tests/integration/models"