Fixed up demo code.