added coverage in tests