Fix issue with migration class naming