diff --git a/keystore2/aidl/Android.bp b/keystore2/aidl/Android.bp
index 00f38aa..3b2dff0 100644
--- a/keystore2/aidl/Android.bp
+++ b/keystore2/aidl/Android.bp
@@ -17,7 +17,7 @@
     srcs: [
         "android/security/attestationmanager/*.aidl",
     ],
-    imports: [ "android.hardware.keymint" ],
+    imports: [ "android.hardware.security.keymint" ],
     unstable: true,
     backend: {
         java: {
diff --git a/keystore2/aidl/android/security/attestationmanager/IAttestationManager.aidl b/keystore2/aidl/android/security/attestationmanager/IAttestationManager.aidl
index d88a298..85eee57 100644
--- a/keystore2/aidl/android/security/attestationmanager/IAttestationManager.aidl
+++ b/keystore2/aidl/android/security/attestationmanager/IAttestationManager.aidl
@@ -17,7 +17,7 @@
 package android.security.attestationmanager;
 
 import android.security.attestationmanager.ByteArray;
-import android.hardware.keymint.KeyParameter;
+import android.hardware.security.keymint.KeyParameter;
 
 /**
  * Internal interface for performing device attestation.