Update unittetst for relationships