Fix relationship bug