Fixing bug in supervisor sample