align code with the website
fix testdrive#wait() in case of a passed number instead of a function