Fix Test after library changes