Catch KeyError, not ValueError