Modify test to pass in travis