Minor change in demo