Fix test following Travis build