fix  missing tests for #4 and #3