1 files changed, 4 insertions, 4 deletions
diff --git a/src/main/java/com/google/devtools/build/lib/packages/DependencyFilter.java b/src/main/java/com/google/devtools/build/lib/packages/DependencyFilter.java
index a5d41ee631..88fe6d4157 100644
--- a/src/main/java/com/google/devtools/build/lib/packages/DependencyFilter.java
+++ b/src/main/java/com/google/devtools/build/lib/packages/DependencyFilter.java
@@ -15,6 +15,7 @@ package com.google.devtools.build.lib.packages;
 
 import com.google.devtools.build.lib.packages.Attribute.ConfigurationTransition;
 import com.google.devtools.build.lib.packages.DependencyFilter.AttributeInfoProvider;
+import com.google.devtools.build.lib.syntax.Type.LabelClass;
 import com.google.devtools.build.lib.util.BinaryPredicate;
 
 /**
@@ -38,8 +39,8 @@ public abstract class DependencyFilter
       new DependencyFilter() {
     @Override
     public boolean apply(AttributeInfoProvider infoProvider, Attribute attribute) {
-      // isHostConfiguration() is only defined for labels and label lists.
-      if (attribute.getType() != BuildType.LABEL && attribute.getType() != BuildType.LABEL_LIST) {
+      // getConfigurationTransition() is only defined for labels which introduce a dependency.
+      if (attribute.getType().getLabelClass() != LabelClass.DEPENDENCY) {
         return true;
       }
 
@@ -62,8 +63,7 @@ public abstract class DependencyFilter
       new DependencyFilter() {
     @Override
     public boolean apply(AttributeInfoProvider infoProvider, Attribute attribute) {
-      return attribute.getType() != BuildType.NODEP_LABEL
-          && attribute.getType() != BuildType.NODEP_LABEL_LIST;
+      return attribute.getType().getLabelClass() != LabelClass.NONDEP_REFERENCE;
     }
   };
   /**