Fix for migrations