Fix for demo controller not always updating the page