Fix test for relationship with inheritance