Fix missing dependency in test files