Fix build

Reviewed By: @ptarjan