Fix yml to make test pass