diff --git a/apps/web/src/components/ChatView.tsx b/apps/web/src/components/ChatView.tsx
index b386430568bef8073872ec9a13a2f4f78051c403..cfc5f81af0f7147119c05aaf770ac51de6c67e69 100644
--- a/apps/web/src/components/ChatView.tsx
+++ b/apps/web/src/components/ChatView.tsx
@@ -93,6 +93,7 @@ import {
   InfoIcon,
   LockIcon,
   LockOpenIcon,
+  Undo2Icon,
   XIcon,
 } from "lucide-react";
 import { Button } from "./ui/button";
@@ -2601,7 +2602,7 @@ const MessagesTimeline = memo(function MessagesTimeline({
                         disabled={isRevertingCheckpoint || isWorking}
                         onClick={() => onRevertUserMessage(timelineEntry.message.id)}
                       >
-                        {isRevertingCheckpoint ? "Reverting..." : "Revert agent work"}
+                        <Undo2Icon className="size-3" />
                       </Button>
                     )}
                     <p className="text-right text-[10px] text-muted-foreground/30">