Add migration in integration tests