It seemed like there was missing logic for if @relationship? Did I miss something, or was this just not implemented?