Update tests to support missing library