another bug fixed... top_k being rewritten