fix unit test and add file