Fix instantiation of migration class