fix for test failure

Fix for test failure in docs.