Fix merge from omni

Signed-off-by: kecinzer <kecinzer@gmail.com>