Update Exit function for test