One more migration fix