Allow class to be passed through to the relationship assertions