diff --git a/web/client/src/library/components/graph/ModelLineage.tsx b/web/client/src/library/components/graph/ModelLineage.tsx
index 5abe15d134..104e86e000 100644
--- a/web/client/src/library/components/graph/ModelLineage.tsx
+++ b/web/client/src/library/components/graph/ModelLineage.tsx
@@ -427,7 +427,7 @@ function GraphControls({ nodes = [] }: { nodes: Node[] }): JSX.Element {
     <div className="px-2 flex items-center text-xs text-neutral-400 @container">
       <div className="contents">
         <Popover
-          className="flex @lg:hidden bg-none border-none py-1"
+          className="flex @lg:hidden bg-none border-none"
           aria-label="Show lineage node details"
         >
           <Popover.Button
diff --git a/web/client/src/library/components/graph/ModelLineageSearch.tsx b/web/client/src/library/components/graph/ModelLineageSearch.tsx
index 2cf644584e..54c72bdfa0 100644
--- a/web/client/src/library/components/graph/ModelLineageSearch.tsx
+++ b/web/client/src/library/components/graph/ModelLineageSearch.tsx
@@ -38,7 +38,7 @@ export default function ModelLineageSearch({
       className={clsx(
         'w-full',
         showSearchInput
-          ? 'block absolute left-0 right-0 z-10 pr-10 bg-dark'
+          ? 'block absolute top-0 left-0 right-0 z-10 pr-10 bg-light dark:bg-dark @[40rem]:items-end @[40rem]:justify-end @[40rem]:flex @[40rem]:static @[40rem]:pr-0'
           : 'items-end justify-end flex',
       )}
     >