diff --git a/0347_top-k-frequent-elements/python3/bucketsort.py b/0347_top-k-frequent-elements/python3/bucketsort.py
index 2d5cb66..3dfd932 100644
--- a/0347_top-k-frequent-elements/python3/bucketsort.py
+++ b/0347_top-k-frequent-elements/python3/bucketsort.py
@@ -36,11 +36,6 @@ class Solution:
                 # Possible that we filled k elements already in this inner
                 # loop, so get out
                 if len(result) == k:
-                    break
-            
-            if len(result) == k:
-                break
+                    return result
             
             j -= 1
-        
-        return result