Fixed model.

Fix relationship between models.