diff --git a/src/OnboardingSPA/components/ImageUploader/components/ImageUploaderWithText/stylesheet.scss b/src/OnboardingSPA/components/ImageUploader/components/ImageUploaderWithText/stylesheet.scss
index 3d39b7e1f..3a85d48e9 100644
--- a/src/OnboardingSPA/components/ImageUploader/components/ImageUploaderWithText/stylesheet.scss
+++ b/src/OnboardingSPA/components/ImageUploader/components/ImageUploaderWithText/stylesheet.scss
@@ -106,6 +106,11 @@
 					flex-direction: column;
 					justify-content: center;
 					margin-left: 18px;
+					max-width: 27vw;
+
+					@media (max-width: #{ ($break-large) }) {
+						width: 60vw;
+					}
 
 					&__filename {
 						margin: 0;
@@ -133,7 +138,7 @@
 						border: none;
 						font-weight: 510;
 						font-style: normal;
-						background-color: #ff0e9099;
+						background-color: transparent;
 						color: var(--nfd-onboarding-primary);
 						font-size: clamp(0.5rem, 0.4091rem + 0.4545vw, 1.5rem);
 
@@ -143,7 +148,7 @@
 						}
 
 						&__icon {
-							fill: rgb(255, 14, 144);
+							fill: var(--nfd-onboarding-primary);
 						}
 					}
 				}
diff --git a/src/OnboardingSPA/components/StartOptions/index.js b/src/OnboardingSPA/components/StartOptions/index.js
index 6991669e4..4f2fd8b0f 100644
--- a/src/OnboardingSPA/components/StartOptions/index.js
+++ b/src/OnboardingSPA/components/StartOptions/index.js
@@ -53,9 +53,9 @@ const StartOptions = ( { questionnaire, oldFlow, options } ) => {
 	};
 	const selectFlow = ( flow ) => {
 		switch ( flow ) {
-			case 'onboarding':
+			case 'sitebuild':
 				return switchFlow( oldFlow );
-			case 'ai':
+			case 'sitegen':
 				return switchFlow( SITEGEN_FLOW );
 			case 'migration':
 				return window.open( migrationUrl, '_blank' );
@@ -68,6 +68,13 @@ const StartOptions = ( { questionnaire, oldFlow, options } ) => {
 			</p>
 			<div className="nfd-onboarding-sitegen-options__container">
 				{ options.map( ( tab, idx ) => {
+					if (
+						tab.flow === SITEGEN_FLOW &&
+						! validateFlow( brandConfig, tab.flow )
+					) {
+						// Do not show the Sitegen AI option if not enabled for the customer
+						return false;
+					}
 					return (
 						<div
 							className="nfd-onboarding-sitegen-options__container__options"
diff --git a/src/OnboardingSPA/steps/TheFork/contents.js b/src/OnboardingSPA/steps/TheFork/contents.js
index b4fee1673..905a7679b 100644
--- a/src/OnboardingSPA/steps/TheFork/contents.js
+++ b/src/OnboardingSPA/steps/TheFork/contents.js
@@ -15,7 +15,7 @@ const getContents = () => {
 					"We'll stay out of your way.",
 					'wp-module-onboarding'
 				),
-				flow: 'onboarding',
+				flow: 'sitebuild',
 			},
 			{
 				title: __( ' Website Creator', 'wp-module-onboarding' ),
@@ -24,7 +24,7 @@ const getContents = () => {
 					'wp-module-onboarding'
 				),
 				span: __( 'AI', 'wp-module-onboarding' ),
-				flow: 'ai',
+				flow: 'sitegen',
 			},
 			{
 				title: __( 'Hire a Pro', 'wp-module-onboarding' ),