fix test after refactoring the library