add a couple of tests to cover the missing part
add a couple of tests to cover the missing part