fix unit test sample