Fix incorrect feature indices