Fix migration in preparation for deployment