Fix the relations models