Minor fix for travis