Fix list index error