Docs fixed

Example code had a wrong variable in docs