Very small tweaks to `select_top_k`.  #142