diff --git a/go/ir/ir.go b/go/ir/ir.go
index 5ec70d78ef..54e71dd647 100644
--- a/go/ir/ir.go
+++ b/go/ir/ir.go
@@ -80,7 +80,7 @@ func (stmt *TrainStmt) GetModelKind() int {
 	if strings.HasPrefix(estimator, "XGB") {
 		return XGBoost
 	}
-	if strings.HasPrefix(estimator, "KMeans") {
+	if strings.HasPrefix(estimator, "KMEANS") {
 		return KMeans
 	}
 	return TensorFlow
diff --git a/go/modelzooserver/modelzooserver.go b/go/modelzooserver/modelzooserver.go
index 0e9c1736e0..582e4c273d 100644
--- a/go/modelzooserver/modelzooserver.go
+++ b/go/modelzooserver/modelzooserver.go
@@ -98,6 +98,7 @@ func StartModelZooServer(port int, dbConnStr string) {
 
 	pb.RegisterModelZooServerServer(grpcServer, &modelZooServer{DB: mysqlConn})
 
+	logger.Infof("SQLFlow Model Zoo started at: %d", port)
 	grpcServer.Serve(lis)
 }