bug fix for #24 - thanks for pointing this out @danikdanik