Fix exit code testing again