Exit code fix