fixing code in doc