Updated migration tool to use latest version of Axon