Fixed missing test coverage