function getCMSFields() { $fields = parent::getCMSFields(); $values = $fields->fieldByname("Values"); if ($values instanceof GridField) { $values->setModelClass("ColoredProductAttributeValue"); } return $fields; }