Update wait.lua