fix #10 as well

Signed-off-by: Igor Gnatenko <i.gnatenko.brain@gmail.com>