Add a unit test to reproduce the bug