Added some test for 100% coverage