You should add a couple of tests for the scenario described in the previous comment too.