Added test coverage hooks