Minor fix for profuction