fixed some test coverage