Small fix in <which> function