fix keyerror bug