Slight bug fix

Array issue; fixed