diff --git a/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py b/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py
index 6df7356..e129080 100644
--- a/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py
+++ b/NeetCodeRoadmap/Arrays_and_Hashing/347_Top_K_Frequent_Elements.py
@@ -11,9 +11,10 @@ def topKFrequent(nums, k):
                         frequency[x] = 1
                 else:
                         frequency[x] += 1
-        for y in frequency:
-                if frequency[y] >= k:
-                        elements.append(y)
+        for i in range(k):
+                maxKey = max(frequency, key=frequency.get)
+                elements.append(maxKey)
+                del frequency[maxKey]
         return elements