Fix "Undefined variable: arg"