updated unit test to catch case