updated unit test for code fix