diff --git a/services/core/java/com/android/server/locksettings/recoverablekeystore/PlatformKeyManager.java b/services/core/java/com/android/server/locksettings/recoverablekeystore/PlatformKeyManager.java
index ef4dbf57430fe..a8b8361b4f03d 100644
--- a/services/core/java/com/android/server/locksettings/recoverablekeystore/PlatformKeyManager.java
+++ b/services/core/java/com/android/server/locksettings/recoverablekeystore/PlatformKeyManager.java
@@ -146,7 +146,6 @@ public class PlatformKeyManager {
     public void regenerate() throws NoSuchAlgorithmException, KeyStoreException {
         int nextId = getGenerationId() + 1;
         generateAndLoadKey(nextId);
-        setGenerationId(nextId);
     }
 
     /**
@@ -219,7 +218,6 @@ public class PlatformKeyManager {
         }
 
         generateAndLoadKey(generationId);
-        mDatabase.setPlatformKeyGenerationId(mUserId, generationId);
     }
 
     /**
@@ -301,6 +299,8 @@ public class PlatformKeyManager {
                     .setBoundToSpecificSecureUserId(mUserId)
                     .build());
 
+        setGenerationId(generationId);
+
         try {
             secretKey.destroy();
         } catch (DestroyFailedException e) {