fix unit test and update docs