Bugfix on migration classes