diff --git a/action/entrypoint.sh b/action/entrypoint.sh
index 0b5b703..1a71e9f 100644
--- a/action/entrypoint.sh
+++ b/action/entrypoint.sh
@@ -42,6 +42,6 @@ fi;
 # Exit with code from the build step.
 #
 
-if [ $USE_EXIT_CODE = true ]; then
+if [[ $USE_EXIT_CODE == true || $TEST_RUNNER_EXIT_CODE -ne 2 ]]; then
 exit $TEST_RUNNER_EXIT_CODE
 fi;