change in tests to cover 100% of the code