Ensure the model is attached to DataSource for relations