fixes

Signed-off-by: Roman Gorbatko <romangorbatko@gmail.com>