Fixes bug in demo update