enhance unit test to catch missed bug