Fix test in travis