small fix - return exit code

small fix - return exit code #607