Error handling fix

Signed-off-by: Alexander Shepetko <a@shepetko.com>