Fix migration 2 script.