Fix up test after doc update from Alex.