Minor bug fix for demo page