Update migration to work E2E