Update migration to work with any customized user model