diff --git a/client/src/components/PromptsTab.tsx b/client/src/components/PromptsTab.tsx
index c0e188c0..fa590558 100644
--- a/client/src/components/PromptsTab.tsx
+++ b/client/src/components/PromptsTab.tsx
@@ -74,6 +74,7 @@ const PromptsTab = ({
         },
         argName,
         value,
+        promptArgs,
       );
     }
   };
diff --git a/client/src/lib/hooks/useCompletionState.ts b/client/src/lib/hooks/useCompletionState.ts
index a63968ae..26b69a27 100644
--- a/client/src/lib/hooks/useCompletionState.ts
+++ b/client/src/lib/hooks/useCompletionState.ts
@@ -84,7 +84,9 @@ export function useCompletionState(
         }));
 
         try {
-          console.log(state);
+          if (context !== undefined) {
+            delete context[argName];
+          }
 
           const values = await handleCompletion(
             ref,