Correct model relationships