Fix test running in travis