fix documentation test based on updated docs