Add some test code to improve code coverage