diff --git a/pkg/controller/studyjobcontroller/studyjob_controller.go b/pkg/controller/studyjobcontroller/studyjob_controller.go
index 2a4bea9874b..226b75d5826 100644
--- a/pkg/controller/studyjobcontroller/studyjob_controller.go
+++ b/pkg/controller/studyjobcontroller/studyjob_controller.go
@@ -693,25 +693,26 @@ func (r *ReconcileStudyJobController) saveModel(c katibapi.ManagerClient, studyI
 		if trial == nil {
 			return fmt.Errorf("Trial %s not found", trialId)
 		}
-		if len(mets) > 0 {
-			smr := &katibapi.SaveModelRequest{
-				Model: &katibapi.ModelInfo{
-					StudyName:  sc.Name,
-					WorkerId:   mls.WorkerId,
-					Parameters: trial.ParameterSet,
-					Metrics:    mets,
-					ModelPath:  sc.Name,
-				},
-				DataSet: &katibapi.DataSetInfo{
-					Name: sc.Name,
-					Path: sc.Name,
-				},
-			}
-			_, err = c.SaveModel(ctx, smr)
-			if err != nil {
-				return err
-			}
-		}
+		// Disable ModelDB
+		//		if len(mets) > 0 {
+		//			smr := &katibapi.SaveModelRequest{
+		//				Model: &katibapi.ModelInfo{
+		//					StudyName:  sc.Name,
+		//					WorkerId:   mls.WorkerId,
+		//					Parameters: trial.ParameterSet,
+		//					Metrics:    mets,
+		//					ModelPath:  sc.Name,
+		//				},
+		//				DataSet: &katibapi.DataSetInfo{
+		//					Name: sc.Name,
+		//					Path: sc.Name,
+		//				},
+		//			}
+		//			_, err = c.SaveModel(ctx, smr)
+		//			if err != nil {
+		//				return err
+		//			}
+		//		}
 	}
 	return nil
 }
diff --git a/test/scripts/run-tests.sh b/test/scripts/run-tests.sh
index 3a0ca517666..c712b3e9e06 100755
--- a/test/scripts/run-tests.sh
+++ b/test/scripts/run-tests.sh
@@ -61,6 +61,7 @@ echo "VERSION ${VERSION}"
 
 sed -i -e "s@image: katib\/vizier-core@image: ${REGISTRY}\/${REPO_NAME}\/vizier-core:${VERSION}@" manifests/vizier/core/deployment.yaml
 sed -i -e "s@image: katib\/vizier-core-rest@image: ${REGISTRY}\/${REPO_NAME}\/vizier-core-rest:${VERSION}@" manifests/vizier/core-rest/deployment.yaml
+sed -i -e "s@image: katib\/katib-ui@image: ${REGISTRY}\/${REPO_NAME}\/katib-ui:${VERSION}@" manifests/vizier/ui/deployment.yaml
 sed -i -e "s@type: NodePort@type: ClusterIP@" -e "/nodePort: 30678/d" manifests/vizier/core/service.yaml
 sed -i -e "s@image: katib\/studyjob-controller@image: ${REGISTRY}\/${REPO_NAME}\/studyjob-controller:${VERSION}@" manifests/studyjobcontroller/studyjobcontroller.yaml
 sed -i -e "s@image: katib\/suggestion-random@image: ${REGISTRY}\/${REPO_NAME}\/suggestion-random:${VERSION}@" manifests/vizier/suggestion/random/deployment.yaml
@@ -68,7 +69,6 @@ sed -i -e "s@image: katib\/suggestion-grid@image: ${REGISTRY}\/${REPO_NAME}\/sug
 sed -i -e "s@image: katib\/suggestion-hyperband@image: ${REGISTRY}\/${REPO_NAME}\/suggestion-hyperband:${VERSION}@" manifests/vizier/suggestion/hyperband/deployment.yaml
 sed -i -e "s@image: katib\/suggestion-bayesianoptimization@image: ${REGISTRY}\/${REPO_NAME}\/suggestion-bayesianoptimization:${VERSION}@" manifests/vizier/suggestion/bayesianoptimization/deployment.yaml
 sed -i -e "s@image: katib\/earlystopping-medianstopping@image: ${REGISTRY}\/${REPO_NAME}\/earlystopping-medianstopping:${VERSION}@" manifests/vizier/earlystopping/medianstopping/deployment.yaml
-sed -i -e "s@image: katib\/katib-frontend@image: ${REGISTRY}\/${REPO_NAME}\/katib-frontend:${VERSION}@" manifests/modeldb/frontend/deployment.yaml
 sed -i -e "31,37d" manifests/vizier/db/deployment.yaml
 
 cat manifests/vizier/core/deployment.yaml