Re-add exit code.

Accidentally removed exit code.