Fix another migration error