fixing the build

Signed-off-by: Igor Fedorenko <igor@ifedorenko.com>