diff --git a/Android.bp b/Android.bp
index 6535da5..d7c9e54 100644
--- a/Android.bp
+++ b/Android.bp
@@ -591,6 +591,9 @@
     visibility: ["//frameworks/opt/net/ike"],
     srcs: [
         "core/java/android/net/annotations/PolicyDirection.java",
+        "core/java/com/android/internal/util/IState.java",
+        "core/java/com/android/internal/util/State.java",
+        "core/java/com/android/internal/util/StateMachine.java",
         "telephony/java/android/telephony/Annotation.java",
     ],
 }