diff --git a/src/app/problems/components/ProblemCard.tsx b/src/app/problems/components/ProblemCard.tsx
index f15dc44..bd881e4 100644
--- a/src/app/problems/components/ProblemCard.tsx
+++ b/src/app/problems/components/ProblemCard.tsx
@@ -1,3 +1,4 @@
+import { Button } from "@/components/ui/button";
 import { Clock, Zap, Users } from "lucide-react";
 
 interface ProblemCardProps {
@@ -81,9 +82,9 @@ const ProblemCard = ({
 
         {/* Quick run button - appears on hover */}
         <div className="absolute bottom-0 right-0 opacity-0 group-hover:opacity-100 transition-all duration-300 transform translate-x-2 group-hover:translate-x-0">
-          <button className="p-2 rounded-lg bg-primary text-primary-foreground">
+          <Button size="p2">
             <Zap className="h-4 w-4" />
-          </button>
+          </Button>
         </div>
       </div>
     </div>
diff --git a/src/components/ui/button.tsx b/src/components/ui/button.tsx
index 3b11e4c..f1e7306 100644
--- a/src/components/ui/button.tsx
+++ b/src/components/ui/button.tsx
@@ -25,6 +25,7 @@ const buttonVariants = cva(
       },
       size: {
         default: "h-10 px-4 py-2",
+        p2: "rounded-md p-2",
         sm: "h-9 rounded-md px-3",
         lg: "h-12 rounded-lg px-8",
         xl: "h-14 rounded-lg px-10 text-base",