Fix class names in migrations/