diff --git a/client/cmd/install.go b/client/cmd/install.go
index 4a0b734..14b8917 100644
--- a/client/cmd/install.go
+++ b/client/cmd/install.go
@@ -22,6 +22,7 @@ import (
 
 	"github.com/google/uuid"
 	"github.com/spf13/cobra"
+	"gorm.io/gorm"
 )
 
 var (
diff --git a/client/lib/lib.go b/client/lib/lib.go
index 6fa1272..3f0def0 100644
--- a/client/lib/lib.go
+++ b/client/lib/lib.go
@@ -811,7 +811,7 @@ func SearchWithOffset(ctx context.Context, db *gorm.DB, query string, limit, off
 
 const SEARCH_RETRY_COUNT = 3
 
-func retryingSearch(ctx context.Context, db *gorm.DB, query string, limit, offset int, currentRetryNum int) ([]*data.HistoryEntry, error) {
+func retryingSearch(ctx context.Context, db *gorm.DB, query string, limit, offset, currentRetryNum int) ([]*data.HistoryEntry, error) {
 	if ctx == nil && query != "" {
 		return nil, fmt.Errorf("lib.Search called with a nil context and a non-empty query (this should never happen)")
 	}