Add a couple of integration tests for migrating