diff --git a/main.py b/main.py
index e1b0f1620..f66751ca0 100644
--- a/main.py
+++ b/main.py
@@ -195,14 +195,13 @@ def prompt_worker(q, server_instance):
             server_instance.last_prompt_id = prompt_id
 
             sensitive = item[5]
-            extra_data = item[3].copy()
-            for k in sensitive:
-                extra_data[k] = sensitive[k]
+            extra_data = {**item[3], **sensitive}
 
             e.execute(item[2], prompt_id, extra_data, item[4])
             need_gc = True
 
-            remove_sensitive = lambda prompt: prompt[:5] + prompt[6:]
+            def remove_sensitive(prompt):
+                return prompt[:5] + prompt[6:]
             q.task_done(item_id,
                         e.history_result,
                         status=execution.PromptQueue.ExecutionStatus(