fix bug and start test unit