Fix for a migration