fix potential KeyError