Corrected demo code.