fixing demo 2