Prevent list index out of range error