fixed test coverage
fixed test coverage