Trying to fix relationship test