diff --git a/archivista/download.go b/archivista/download.go
index a856fb4a..cb6127b8 100644
--- a/archivista/download.go
+++ b/archivista/download.go
@@ -17,8 +17,8 @@ package archivista
 import (
 	"context"
 
+	archivistaapi "github.com/in-toto/archivista/pkg/api"
 	"github.com/in-toto/go-witness/dsse"
-	archivistaapi "github.com/testifysec/archivista-api"
 )
 
 func (c *Client) Download(ctx context.Context, gitoid string) (dsse.Envelope, error) {
diff --git a/archivista/searchgitoids.go b/archivista/searchgitoids.go
index 851840dc..f52d5257 100644
--- a/archivista/searchgitoids.go
+++ b/archivista/searchgitoids.go
@@ -17,7 +17,7 @@ package archivista
 import (
 	"context"
 
-	archivistaapi "github.com/testifysec/archivista-api"
+	archivistaapi "github.com/in-toto/archivista/pkg/api"
 )
 
 type searchGitoidResponse struct {
diff --git a/archivista/store.go b/archivista/store.go
index e79daab0..ccba927a 100644
--- a/archivista/store.go
+++ b/archivista/store.go
@@ -17,8 +17,8 @@ package archivista
 import (
 	"context"
 
+	archivistaapi "github.com/in-toto/archivista/pkg/api"
 	"github.com/in-toto/go-witness/dsse"
-	archivistaapi "github.com/testifysec/archivista-api"
 )
 
 func (c *Client) Store(ctx context.Context, env dsse.Envelope) (string, error) {