Fix old migration

This MR fixes a problem we had when running the 0740d341ea8e_.py on our staging system. I'm not sure if this is the only problem that we have on staging, but fixing it will be important for the production release.