Migration bug fix

not sure what broke but it's fixed.