Fix javascript for baseline demo