diff --git a/android/src/main/java/com/facebook/sonar/plugins/litho/DebugComponentDescriptor.java b/android/src/main/java/com/facebook/sonar/plugins/litho/DebugComponentDescriptor.java
index 8267bda05..83771b1f0 100644
--- a/android/src/main/java/com/facebook/sonar/plugins/litho/DebugComponentDescriptor.java
+++ b/android/src/main/java/com/facebook/sonar/plugins/litho/DebugComponentDescriptor.java
@@ -19,6 +19,7 @@ import com.facebook.litho.ComponentLifecycle;
 import com.facebook.litho.DebugComponent;
 import com.facebook.litho.DebugLayoutNode;
 import com.facebook.litho.LithoView;
+import com.facebook.litho.StateContainer;
 import com.facebook.litho.annotations.Prop;
 import com.facebook.litho.annotations.State;
 import com.facebook.litho.reference.Reference;
@@ -71,7 +72,7 @@ public class DebugComponentDescriptor extends NodeDescriptor<DebugComponent> {
 
         @Override
         public void applyStateOverrides(
-            String key, ComponentLifecycle.StateContainer stateContainer) {
+            String key, StateContainer stateContainer) {
           final List<Pair<String[], SonarDynamic>> overrides = mOverrides.get(key);
           if (overrides == null) {
             return;
@@ -421,7 +422,7 @@ public class DebugComponentDescriptor extends NodeDescriptor<DebugComponent> {
       return null;
     }
 
-    final ComponentLifecycle.StateContainer stateContainer = node.getStateContainer();
+    final StateContainer stateContainer = node.getStateContainer();
     if (stateContainer == null) {
       return null;
     }