add logic to run migration on end of the setup