Fix error to demo