Fixing example to work in run mode