Fix bug in docs code.