fixed highlighting issue when user changed demo code