Correction of demo code.