small migration fix