Add test unit, see the issue #6