You are right, there are 2 cases - I will add more tests.