Slight fix in code example.

Signed-off-by: Shwetank Dixit <emailshwetank@gmail.com>