add at_exit for CI