Fix change in SDK upgrade

Azure · Apr 14, 2021 · e4fca8a · e4fca8a
1 parent 1934bcf
commit e4fca8a
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/src/Storage/Storage.Management/Models/PSStorageAccount.cs b/src/Storage/Storage.Management/Models/PSStorageAccount.cs
@@ -238,18 +238,12 @@ public class PSKeyCreationTime
         public PSKeyCreationTime()
         { }
 
-        public PSKeyCreationTime(IDictionary<string, DateTime?> keyCreationTime)
+        public PSKeyCreationTime(KeyCreationTime keyCreationTime)
         {
-            if (keyCreationTime != null && keyCreationTime.Count > 0)
+            if (keyCreationTime != null)
             {
-                if (keyCreationTime.Keys.Contains("key1"))
-                {
-                    Key1 = keyCreationTime["key1"];
-                }
-                if (keyCreationTime.Keys.Contains("key2"))
-                {
-                    Key2 = keyCreationTime["key2"];
-                }
+                this.Key1 = keyCreationTime.Key1;
+                this.Key2 = keyCreationTime.Key2;
             }
         }
         public System.DateTime? Key1 { get; set; }