Update issue6(#12) unitest
Fix issue6