fix code sample in doc