diff --git a/cmd/cmd.go b/cmd/cmd.go
index 57797e931..ce7187aee 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -626,7 +626,7 @@ func PushHandler(cmd *cobra.Command, args []string) error {
 	spinner.Stop()
 
 	destination := n.String()
-	if strings.HasSuffix(n.Host, ".ollama.ai") || strings.HasSuffix(n.Host, ".ollama.com") {
+	if isOllamaHost {
 		destination = "https://ollama.com/" + strings.TrimSuffix(n.DisplayShortest(), ":latest")
 	}
 	fmt.Printf("\nYou can find your model at:\n\n")