This is used in multiple places, so it has a little more changes than
the previous ones.

Signed-off-by: Iustin Pop <iustin@google.com>
Reviewed-by: Guido Trotter <ultrotter@google.com>