Fix model's functionality attribute issue