fix array index error