Add small fixes

Signed-off-by: Igor Popov <igor@igorpopov.io>