Fix test env KeyError