Fix the exit bug
Fix the exit bug