Add missing dependency for tests