fixed test coverage issues