Small fix

Array out of bounds error at if statement at line 96.