Ensure models from relations are included