Added integration tests for the migration helper