bug fixes

Signed-off-by: Marcin Wiatr <mail.wichura@gmail.com>
updates

Signed-off-by: Marcin Wiatr <mail.wichura@gmail.com>