Fixed code example in docs