getting better test converage :smile: