diff --git a/gradio/cli/commands/components/_docs_utils.py b/gradio/cli/commands/components/_docs_utils.py
index 76744f1b93..420f9f8fc5 100644
--- a/gradio/cli/commands/components/_docs_utils.py
+++ b/gradio/cli/commands/components/_docs_utils.py
@@ -9,9 +9,9 @@
 
 def find_first_non_return_key(some_dict):
     """Finds the first key in a dictionary that is not "return"."""
-    for key, value in some_dict.items():
+    for key in some_dict:
         if key != "return":
-            return value
+            return some_dict[key]
     return None