Migration Fix

This should fix the migration issue on the Heroku side.