Replace os.exit with sys.exit