Small bug fix with ttesting and code coverage