Fix one test because it was failing in travis