BOOK-719 fixing tests