Fix test on Travis