Added unit test for previous minor fix