fix the array bug