#127 unit test and fix

[amihaiemil]

PR for #127

[zoeself]

@amihaiemil thank you for your Pull Request. I'll assign someone to review it soon.

[zoeself]

@amihaiemil I couldn't find any assignee for this task. This is either because there are no contributors with role REV available or because the project does not have enough funds.

Please, make sure there is at least one available contributor with the required role and the project can afford to pay them.