[FIX]: minor fix to the code block