Add test coverage back.