Fix exit code