Update unittest for relationships