Add a unit test for the bug