Add missing relationship