Make changes to pyro_cluster code so as it works.