Fix migrations for production