Fix models relations