Fix relationship model error