fix code sample in docs