Fix the lacking dependency in tests