Fix: A minor fix in the demonstration script.