Fix IndexOutOfBoundsException