Fix: A minor fix to the demonstration script.