fix error in demo