Fix bug in regression demo.