add coverage in tests