fixes bug in demo task