Adjust models' associations