Fix confused demos 3 and 4