Add unit test for more condition