Fixed it for the model extension.