4 files changed, 13 insertions, 10 deletions

diff --git a/go/cmd/gitaly-receive-pack/main.go b/go/cmd/gitaly-receive-pack/main.go
index 06b8a72..511d584 100644
--- a/go/cmd/gitaly-receive-pack/main.go
+++ b/go/cmd/gitaly-receive-pack/main.go
@@ -5,9 +5,10 @@ import (
 	"fmt"
 	"os"
 
-	"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/handler"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/logger"
+
+	pb "gitlab.com/gitlab-org/gitaly-proto/go"
 )
 
 func init() {
@@ -23,7 +24,7 @@ func main() {
 		logger.Fatal("wrong number of arguments", fmt.Errorf("expected 2 arguments, got %v", os.Args))
 	}
 
-	var request gitalypb.SSHReceivePackRequest
+	var request pb.SSHReceivePackRequest
 	if err := json.Unmarshal([]byte(os.Args[2]), &request); err != nil {
 		logger.Fatal("unmarshaling request json failed", err)
 	}
diff --git a/go/cmd/gitaly-upload-archive/main.go b/go/cmd/gitaly-upload-archive/main.go
index 845f8f4..c988baa 100644
--- a/go/cmd/gitaly-upload-archive/main.go
+++ b/go/cmd/gitaly-upload-archive/main.go
@@ -5,16 +5,17 @@ import (
 	"fmt"
 	"os"
 
-	"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/handler"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/logger"
+
+	pb "gitlab.com/gitlab-org/gitaly-proto/go"
 )
 
 func init() {
 	logger.ProgName = "gitaly-upload-archive"
 }
 
-type uploadArchiveHandler func(gitalyAddress string, request *gitalypb.SSHUploadArchiveRequest) (int32, error)
+type uploadArchiveHandler func(gitalyAddress string, request *pb.SSHUploadArchiveRequest) (int32, error)
 
 func main() {
 	if err := handler.Prepare(); err != nil {
@@ -35,7 +36,7 @@ func uploadArchive(handler uploadArchiveHandler, args []string) (int32, error) {
 		return 0, fmt.Errorf("wrong number of arguments: expected 2 arguments, got %v", args)
 	}
 
-	var request gitalypb.SSHUploadArchiveRequest
+	var request pb.SSHUploadArchiveRequest
 	if err := json.Unmarshal([]byte(args[2]), &request); err != nil {
 		return 0, fmt.Errorf("unmarshaling request json failed: %v", err)
 	}
diff --git a/go/cmd/gitaly-upload-archive/main_test.go b/go/cmd/gitaly-upload-archive/main_test.go
index 7cb9942..b17da7b 100644
--- a/go/cmd/gitaly-upload-archive/main_test.go
+++ b/go/cmd/gitaly-upload-archive/main_test.go
@@ -5,7 +5,7 @@ import (
 	"strings"
 	"testing"
 
-	"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
+	pb "gitlab.com/gitlab-org/gitaly-proto/go"
 )
 
 var testGitalyAddress = "unix:gitaly.socket"
@@ -13,7 +13,7 @@ var testGitalyAddress = "unix:gitaly.socket"
 func TestUploadArchiveSuccess(t *testing.T) {
 	testRelativePath := "myrepo.git"
 	requestJSON := fmt.Sprintf(`{"repository":{"relative_path":"%s"}}`, testRelativePath)
-	mockHandler := func(gitalyAddress string, request *gitalypb.SSHUploadArchiveRequest) (int32, error) {
+	mockHandler := func(gitalyAddress string, request *pb.SSHUploadArchiveRequest) (int32, error) {
 		if gitalyAddress != testGitalyAddress {
 			t.Fatalf("Expected gitaly address %s got %v", testGitalyAddress, gitalyAddress)
 		}
@@ -35,7 +35,7 @@ func TestUploadArchiveSuccess(t *testing.T) {
 }
 
 func TestUploadArchiveFailure(t *testing.T) {
-	mockHandler := func(_ string, _ *gitalypb.SSHUploadArchiveRequest) (int32, error) {
+	mockHandler := func(_ string, _ *pb.SSHUploadArchiveRequest) (int32, error) {
 		t.Fatal("Expected handler not to be called")
 
 		return 0, nil
diff --git a/go/cmd/gitaly-upload-pack/main.go b/go/cmd/gitaly-upload-pack/main.go
index a2250b2..282c90f 100644
--- a/go/cmd/gitaly-upload-pack/main.go
+++ b/go/cmd/gitaly-upload-pack/main.go
@@ -5,9 +5,10 @@ import (
 	"fmt"
 	"os"
 
-	"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/handler"
 	"gitlab.com/gitlab-org/gitlab-shell/go/internal/logger"
+
+	pb "gitlab.com/gitlab-org/gitaly-proto/go"
 )
 
 func init() {
@@ -23,7 +24,7 @@ func main() {
 		logger.Fatal("wrong number of arguments", fmt.Errorf("expected 2 arguments, got %v", os.Args))
 	}
 
-	var request gitalypb.SSHUploadPackRequest
+	var request pb.SSHUploadPackRequest
 	if err := json.Unmarshal([]byte(os.Args[2]), &request); err != nil {
 		logger.Fatal("unmarshaling request json failed", err)
 	}