Fix in travis unittest