Add more code coverage