Correct class name of migration