1 files changed, 8 insertions, 5 deletions

diff --git a/packages/renderer/src/components/locationBar/LocationTextField.tsx b/packages/renderer/src/components/locationBar/LocationTextField.tsx
index 30953de..85cf794 100644
--- a/packages/renderer/src/components/locationBar/LocationTextField.tsx
+++ b/packages/renderer/src/components/locationBar/LocationTextField.tsx
@@ -20,6 +20,7 @@
 
 import FilledInput from '@mui/material/FilledInput';
 import { styled } from '@mui/material/styles';
+import { SecurityLabelKind } from '@sophie/shared';
 import { autorun } from 'mobx';
 import { observer } from 'mobx-react-lite';
 import React, { useCallback, useEffect, useState } from 'react';
@@ -30,7 +31,6 @@ import GoButton from './GoButton';
 import LocationOverlayInput from './LocationOverlayInput';
 import SecurityLabel from './SecurityLabel';
 import UrlOverlay from './UrlOverlay';
-import splitUrl from './splitUrl';
 
 const LocationTextFieldRoot = styled(FilledInput, {
   name: 'LocationTextField',
@@ -75,8 +75,6 @@ function LocationTextField({
     [setInputFocused],
   );
 
-  const splitResult = splitUrl(service?.currentUrl);
-
   return (
     <LocationTextFieldRoot
       inputComponent={LocationOverlayInput}
@@ -84,7 +82,12 @@ function LocationTextField({
         'aria-label': 'Location',
         spellCheck: false,
         overlayVisible: !inputFocused && !changed,
-        overlay: <UrlOverlay splitResult={splitResult} />,
+        overlay: (
+          <UrlOverlay
+            url={service?.currentUrl ?? ''}
+            alert={service?.hasSecurityLabelWarning ?? false}
+          />
+        ),
       }}
       inputRef={inputRefCallback}
       onFocus={() => setInputFocused(true)}
@@ -114,8 +117,8 @@ function LocationTextField({
       disableUnderline
       startAdornment={
         <SecurityLabel
+          kind={service?.securityLabel ?? SecurityLabelKind.Empty}
           changed={changed}
-          splitResult={splitResult}
           position="start"
         />
       }