fixed class name for migration >.<