Fix unit test missing dependency