Fixed something I forgot in the relationships