Travis build error fix