fix for Migrate class name