coverage fix for tests