keyerror bug fix