fix yml for pregel test