IndexOutOfBoundsException fix