diff --git a/components/agents/AgentGenerationWorkspaceDarkV2.tsx b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
index c06f28a05..82462d754 100644
--- a/components/agents/AgentGenerationWorkspaceDarkV2.tsx
+++ b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
@@ -1469,7 +1469,7 @@ export function AgentGenerationWorkspaceDarkV2({
       </div>
       {agentUploadSlots.length > 0 ? (
         <div className="space-y-4">
-          <div className="grid grid-cols-1 gap-3 md:grid-cols-3">
+          <div className="grid grid-cols-2 gap-3 lg:grid-cols-4">
             {agentUploadSlots.map((slot, index) => {
               const isActiveAssignmentSlot = activeAssignmentSlotIndex === index;
               return (
@@ -1568,7 +1568,7 @@ export function AgentGenerationWorkspaceDarkV2({
               </div>
 
               {assignableSourceLibrary.length > 0 ? (
-                <div className="grid grid-cols-1 gap-3 md:grid-cols-3">
+                <div className="grid grid-cols-2 gap-3 lg:grid-cols-4">
                   {assignableSourceLibrary.map((source) => {
                     const assignedToActiveSlot =
                       activeAssignmentSlotIndex !== null &&