diff --git a/test/keys_only/keys_only.py b/test/keys_only/keys_only.py
index 43d879917..ab1f21176 100644
--- a/test/keys_only/keys_only.py
+++ b/test/keys_only/keys_only.py
@@ -1,5 +1,2 @@
 def keys_only(flat_dict):
-    lst = []
-    for k, v in flat_dict.items():
-        lst.append(k)
-    return lst
\ No newline at end of file
+    return list(flat_dict.keys())
\ No newline at end of file