Allow migration script to run correctly on tests.