Fix a layout bug in demo.