fix migeration bugs