Modify top-k selection, revert back