Fix exit code

Resolve #5