Fix merge artifacts in demo notebook.