After extending the key vocabulary, these output layers need to be renamed all throughout (e.g., unit tests).