diff --git a/default_api_test.go b/default_api_test.go
index 9350e0a..2dc13e7 100644
--- a/default_api_test.go
+++ b/default_api_test.go
@@ -2288,10 +2288,10 @@ func TestDefaultApiService_GetArchive(t *testing.T) {
 		name                     string
 		fields                   fields
 		args                     args
-		want                     *APIResponse
+		want                     int64
 		wantErr, integrationTest bool
 	}{
-		{"networkErrorContextExceeded", fields{client: generateConfigFake()}, args{}, &APIResponse{Message: "Get https://stash.domain.com/rest/api/1.0/projects//repos//archive: context canceled"}, true, false},
+		{"networkErrorContextExceeded", fields{client: generateConfigFake()}, args{}, 0, true, false},
 	}
 	for _, tt := range tests {
 		if tt.integrationTest != runIntegrationTests {