Added stuff for code coverage