Try to fix the test coverage