Updated the example to match onTestComplete's new behaviour.

Related to / fixes #116 #137