Corrected code in docs