diff --git a/openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/QueryBuilderWidget/QueryBuilderWidget.tsx b/openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/QueryBuilderWidget/QueryBuilderWidget.tsx
index 61d9362d3dd..42b703adff2 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/QueryBuilderWidget/QueryBuilderWidget.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/common/Form/JSONSchema/JsonSchemaWidgets/QueryBuilderWidget/QueryBuilderWidget.tsx
@@ -46,10 +46,10 @@ const QueryBuilderWidget: FC<WidgetProps> = ({
     useAdvanceSearch();
   const [searchResults, setSearchResults] = useState<number>(0);
   const entityType =
-    (props.formContext?.entityType ?? props?.entityType) || EntityType.ALL;
+    (props.formContext?.entityType ?? schema?.entityType) || EntityType.ALL;
   const searchIndexMapping = searchClassBase.getEntityTypeSearchIndexMapping();
   const searchIndex = searchIndexMapping[entityType as string];
-  const outputType = props?.outputType ?? QueryBuilderOutputType.ELASTICSEARCH;
+  const outputType = schema?.outputType ?? QueryBuilderOutputType.ELASTICSEARCH;
 
   const fetchEntityCount = useCallback(
     async (queryFilter: Record<string, unknown>) => {
@@ -152,17 +152,17 @@ const QueryBuilderWidget: FC<WidgetProps> = ({
                 showIcon
                 icon={<InfoCircleOutlined height={16} />}
                 message={
-                  <>
+                  <div className="d-flex flex-wrap items-center gap-1">
                     <Typography.Text>
                       {t('message.search-entity-count', {
                         count: searchResults,
                       })}
                     </Typography.Text>
 
-                    <Typography.Text className="m-l-sm text-xs text-grey-muted">
+                    <Typography.Text className="text-xs text-grey-muted">
                       {t('message.click-here-to-view-assets-on-explore')}
                     </Typography.Text>
-                  </>
+                  </div>
                 }
                 type="info"
               />
diff --git a/openmetadata-ui/src/main/resources/ui/src/pages/AppInstall/AppInstall.component.tsx b/openmetadata-ui/src/main/resources/ui/src/pages/AppInstall/AppInstall.component.tsx
index 60eabf8ce79..9a303333f06 100644
--- a/openmetadata-ui/src/main/resources/ui/src/pages/AppInstall/AppInstall.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/pages/AppInstall/AppInstall.component.tsx
@@ -193,7 +193,7 @@ const AppInstall = () => {
 
       case 2:
         return (
-          <div className="w-500 p-md border rounded-4">
+          <div className="w-4/5 p-md border rounded-4">
             <FormBuilder
               showFormHeader
               useSelectWidget