minor refactor
minor change
minor refactor
add the new test case