Fix bad_index argument bug