Fix test callbacks (teardown_all -> on_exit)

As requested on #121.