diff --git a/front/components/vaults/AddToVaultDialog.tsx b/front/components/vaults/AddToVaultDialog.tsx
index bd18b26b5dea..e2e0f84921ff 100644
--- a/front/components/vaults/AddToVaultDialog.tsx
+++ b/front/components/vaults/AddToVaultDialog.tsx
@@ -1,4 +1,4 @@
-import { Button, Dialog, DropdownMenu, LockIcon } from "@dust-tt/sparkle";
+import { Button, Dialog, DropdownMenu } from "@dust-tt/sparkle";
 import type {
   APIError,
   DataSourceViewContentNode,
@@ -142,10 +142,11 @@ export const AddToVaultDialog = ({
         <DropdownMenu>
           <DropdownMenu.Button>
             <Button
+              hasMagnifying={false}
               label={vault ? vault.name : "Select vault"}
-              variant="primary"
-              icon={LockIcon}
               size="sm"
+              type="select"
+              variant="secondary"
             />
           </DropdownMenu.Button>
           <DropdownMenu.Items>