[MINOR] Added a missing curly braces..

Just a missing curly braces to make the sample valid..

Thanks for making this !