Add support for labels defined in the model