diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
index da635021a..51b1f3bb9 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -484,7 +484,7 @@ def run_ai_cell(self, args: CellArgs, prompt: str):
             # Get the output, either as raw text or as the contents of the 'text' key of a dict
             invoke_output = self.custom_model_registry[args.model_id].invoke(prompt)
             if isinstance(invoke_output, dict):
-                invoke_output = invoke_output.get('text')
+                invoke_output = invoke_output.get("text")
 
             return self.display_output(
                 invoke_output,