[update] minor adjustment to fit into our needs.

This reverts commit c556230378dff8382f5a10848b4778044e40b285.