Fix for test in travis