diff --git a/datahub-web-react/src/app/ingest/source/builder/NameSourceStep.tsx b/datahub-web-react/src/app/ingest/source/builder/NameSourceStep.tsx
index 53a66f9cca..0290b5595e 100644
--- a/datahub-web-react/src/app/ingest/source/builder/NameSourceStep.tsx
+++ b/datahub-web-react/src/app/ingest/source/builder/NameSourceStep.tsx
@@ -1,5 +1,6 @@
 import { Checkbox, Collapse, Form, Input, Typography } from 'antd';
 import { Button, Tooltip } from '@components';
+import { ModalButtonContainer } from '@src/app/shared/button/styledComponents';
 import React from 'react';
 import styled from 'styled-components';
 import { SourceBuilderState, StepProps, StringMapEntryInput } from './types';
@@ -11,10 +12,6 @@ const ControlsContainer = styled.div`
     margin-top: 8px;
 `;
 
-const SaveButton = styled(Button)`
-    margin-right: 15px;
-`;
-
 const ExtraEnvKey = 'extra_env_vars';
 const ExtraReqKey = 'extra_pip_requirements';
 const ExtraPluginKey = 'extra_pip_plugins';
@@ -259,15 +256,15 @@ export const NameSourceStep = ({ state, updateState, prev, submit }: StepProps)
                 <Button variant="outline" color="gray" onClick={prev}>
                     Previous
                 </Button>
-                <div>
-                    <SaveButton
+                <ModalButtonContainer>
+                    <Button
                         variant="outline"
                         data-testid="ingestion-source-save-button"
                         disabled={!(state.name !== undefined && state.name.length > 0)}
                         onClick={() => onClickCreate(false)}
                     >
                         Save
-                    </SaveButton>
+                    </Button>
                     <Tooltip showArrow={false} title="Save and starting syncing data source">
                         <Button
                             disabled={!(state.name !== undefined && state.name.length > 0)}
@@ -276,7 +273,7 @@ export const NameSourceStep = ({ state, updateState, prev, submit }: StepProps)
                             Save & Run
                         </Button>
                     </Tooltip>
-                </div>
+                </ModalButtonContainer>
             </ControlsContainer>
         </>
     );
diff --git a/datahub-web-react/src/app/ingest/source/builder/RecipeForm/TestConnection/TestConnectionButton.tsx b/datahub-web-react/src/app/ingest/source/builder/RecipeForm/TestConnection/TestConnectionButton.tsx
index 3062c2ad26..6ea7db4d71 100644
--- a/datahub-web-react/src/app/ingest/source/builder/RecipeForm/TestConnection/TestConnectionButton.tsx
+++ b/datahub-web-react/src/app/ingest/source/builder/RecipeForm/TestConnection/TestConnectionButton.tsx
@@ -108,7 +108,7 @@ function TestConnectionButton(props: Props) {
 
     return (
         <>
-            <Button variant="outline" onClick={testConnection}>
+            <Button variant="outline" type="button" onClick={testConnection}>
                 <CheckCircleOutlined />
                 Test Connection
             </Button>
diff --git a/datahub-web-react/src/app/ingest/source/builder/constants.ts b/datahub-web-react/src/app/ingest/source/builder/constants.ts
index 9711dfe2f0..73a100c3c5 100644
--- a/datahub-web-react/src/app/ingest/source/builder/constants.ts
+++ b/datahub-web-react/src/app/ingest/source/builder/constants.ts
@@ -39,6 +39,7 @@ import sacLogo from '../../../../images/saclogo.svg';
 import cassandraLogo from '../../../../images/cassandralogo.png';
 import datahubLogo from '../../../../images/datahublogo.png';
 import neo4j from '../../../../images/neo4j.png';
+import vertexAI from '../../../../images/vertexai.png';
 
 export const ATHENA = 'athena';
 export const ATHENA_URN = `urn:li:dataPlatform:${ATHENA}`;
@@ -140,6 +141,8 @@ export const DATAHUB_BUSINESS_GLOSSARY = 'datahub-business-glossary';
 export const DATAHUB_URN = `urn:li:dataPlatform:${DATAHUB}`;
 export const NEO4J = 'neo4j';
 export const NEO4J_URN = `urn:li:dataPlatform:${NEO4J}`;
+export const VERTEX_AI = 'vertexai';
+export const VERTEXAI_URN = `urn:li:dataPlatform:${VERTEX_AI}`;
 
 export const PLATFORM_URN_TO_LOGO = {
     [ATHENA_URN]: athenaLogo,
@@ -184,6 +187,7 @@ export const PLATFORM_URN_TO_LOGO = {
     [CASSANDRA_URN]: cassandraLogo,
     [DATAHUB_URN]: datahubLogo,
     [NEO4J_URN]: neo4j,
+    [VERTEXAI_URN]: vertexAI,
 };
 
 export const SOURCE_TO_PLATFORM_URN = {
diff --git a/datahub-web-react/src/app/ingest/source/builder/sources.json b/datahub-web-react/src/app/ingest/source/builder/sources.json
index c288221483..45770ec0f1 100644
--- a/datahub-web-react/src/app/ingest/source/builder/sources.json
+++ b/datahub-web-react/src/app/ingest/source/builder/sources.json
@@ -337,7 +337,8 @@
     {
         "urn": "urn:li:dataPlatform:vertexai",
         "name": "vertexai",
-        "displayName": "VertexAI",
+        "displayName": "Vertex AI",
+        "description": "Import ML Models and lineage from Google Vertex AI.",
         "docsUrl": "https://datahubproject.io/docs/generated/ingestion/sources/vertexai/",
         "recipe": "source:\n  type: vertexai\n  config:\n    project_id: # you GCP project ID \n     region: # region where your GCP project resides  \n   # Credentials\n        # Add GCP credentials"
     }