Minor fix to use label instead of string