Image migration rollback fix