Fixed code bug in doc