Tests should now end the application

Thought I added this a while ago, but it's in there now.