Minor fix for entities relationships