[r=julian-edwards][bug=1359550,1359551][author=julian-edwards] Fix a migration error that was a) not importing an exception that it was using, b) not catching all the exceptions it should be.
Also fix the code where it was copied from.