Reverted custom migration code