Fix missing relationship