Minor fix for ubuntu