Minor fix to get Travis working