Adding relationship test not working perfectly yet