Fixing a bug with quit