Fix migrations to use generic user model