Use the custom #attribute_type method everywhere