found another KeyError, fixed