Good catch, will add tests for it aswell.