Fix Code Coverage