diff --git a/test_context.go b/test_context.go
index 2d9fe3f..e88225e 100644
--- a/test_context.go
+++ b/test_context.go
@@ -13,8 +13,8 @@ import (
 	"github.com/cucumber/godog/internal/models"
 )
 
-// Feature represents gherkin document.
-type Feature = messages.GherkinDocument
+// GherkinDocument represents gherkin document.
+type GherkinDocument = messages.GherkinDocument
 
 // Scenario represents the executed scenario
 type Scenario = messages.Pickle