Added one more unit test