Removed legacy code from migration script