toolbox: Fix at exit callback