Added hooks for code coverage