Fix index bug

Signed-off-by: Itai Koren <itkoren@gmail.com>