diff --git a/README.md b/README.md
index 4224d6c..30efeab 100644
--- a/README.md
+++ b/README.md
@@ -30,7 +30,7 @@ import (
 	"context"
 	"os"
 
-	"github.com/replicate/replicate-go/pkg/replicate"
+	"github.com/replicate/replicate-go/replicate"
 )
 
 client := replicate.NewClient(os.Getenv("REPLICATE_API_TOKEN"))
diff --git a/pkg/replicate/apierror.go b/replicate/apierror.go
similarity index 100%
rename from pkg/replicate/apierror.go
rename to replicate/apierror.go
diff --git a/pkg/replicate/client.go b/replicate/client.go
similarity index 100%
rename from pkg/replicate/client.go
rename to replicate/client.go
diff --git a/pkg/replicate/collection.go b/replicate/collection.go
similarity index 100%
rename from pkg/replicate/collection.go
rename to replicate/collection.go
diff --git a/pkg/replicate/model.go b/replicate/model.go
similarity index 100%
rename from pkg/replicate/model.go
rename to replicate/model.go
diff --git a/pkg/replicate/paginate.go b/replicate/paginate.go
similarity index 100%
rename from pkg/replicate/paginate.go
rename to replicate/paginate.go
diff --git a/pkg/replicate/prediction.go b/replicate/prediction.go
similarity index 100%
rename from pkg/replicate/prediction.go
rename to replicate/prediction.go
diff --git a/pkg/replicate/replicate_test.go b/replicate/replicate_test.go
similarity index 99%
rename from pkg/replicate/replicate_test.go
rename to replicate/replicate_test.go
index 4b7fb12..075835e 100644
--- a/pkg/replicate/replicate_test.go
+++ b/replicate/replicate_test.go
@@ -9,7 +9,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/replicate/replicate-go/pkg/replicate"
+	"github.com/replicate/replicate-go/replicate"
 	"github.com/stretchr/testify/assert"
 )
 
diff --git a/pkg/replicate/run.go b/replicate/run.go
similarity index 100%
rename from pkg/replicate/run.go
rename to replicate/run.go
diff --git a/pkg/replicate/status.go b/replicate/status.go
similarity index 100%
rename from pkg/replicate/status.go
rename to replicate/status.go
diff --git a/pkg/replicate/training.go b/replicate/training.go
similarity index 100%
rename from pkg/replicate/training.go
rename to replicate/training.go
diff --git a/pkg/replicate/webhook.go b/replicate/webhook.go
similarity index 100%
rename from pkg/replicate/webhook.go
rename to replicate/webhook.go