Adding some test coverage