fix IndexError