Add unit test for latest bug