diff --git a/src/main/java/com/google/devtools/build/lib/remote/RemoteOutputChecker.java b/src/main/java/com/google/devtools/build/lib/remote/RemoteOutputChecker.java
index a09bc94..c06e7e2 100644
--- a/src/main/java/com/google/devtools/build/lib/remote/RemoteOutputChecker.java
+++ b/src/main/java/com/google/devtools/build/lib/remote/RemoteOutputChecker.java
@@ -194,15 +194,15 @@
     }
   }
 
-  private boolean shouldDownloadOutputForToplevel(ActionInput output) {
-    return shouldDownloadOutputFor(output, toplevelArtifactsToDownload);
+  private boolean isTopLevelArtifact(ActionInput output) {
+    return isPartOfCollectedSet(output, toplevelArtifactsToDownload);
   }
 
-  private boolean shouldDownloadOutputForLocalAction(ActionInput output) {
-    return shouldDownloadOutputFor(output, inputsToDownload);
+  private boolean isInputToLocalAction(ActionInput output) {
+    return isPartOfCollectedSet(output, inputsToDownload);
   }
 
-  private boolean shouldDownloadOutputForRegex(ActionInput output) {
+  private boolean matchesRegex(ActionInput output) {
     if (output instanceof Artifact && ((Artifact) output).isTreeArtifact()) {
       return false;
     }
@@ -216,26 +216,19 @@
     return false;
   }
 
-  private static boolean shouldDownloadOutputFor(
-      ActionInput output, Set<ActionInput> artifactCollection) {
-    if (output instanceof TreeFileArtifact) {
-      if (artifactCollection.contains(((Artifact) output).getParent())) {
-        return true;
-      }
-    } else if (artifactCollection.contains(output)) {
-      return true;
-    }
-
-    return false;
+  private static boolean isPartOfCollectedSet(
+      ActionInput actionInput, Set<ActionInput> artifactSet) {
+    return artifactSet.contains(
+        actionInput instanceof TreeFileArtifact
+            ? ((Artifact) actionInput).getParent()
+            : actionInput);
   }
 
   /**
    * Returns {@code true} if Bazel should download this {@link ActionInput} during spawn execution.
    */
   public boolean shouldDownloadOutput(ActionInput output) {
-    return shouldDownloadOutputForToplevel(output)
-        || shouldDownloadOutputForLocalAction(output)
-        || shouldDownloadOutputForRegex(output);
+    return isTopLevelArtifact(output) || isInputToLocalAction(output) || matchesRegex(output);
   }
 
   @Override