Minor bug fix in the sample.