fixing code in docs