Fix bug with reflect index