fix compilation error

Reviewer: @poolakiran