minor change to PS demo