add unit test and fix broken unit test link