Added 1 more unit test