Fix label wrapper function