Amend the unittest due to more on_exit being triggered