Accidentally unchecked in last commit for types

Author: begoldsm

src/ResourceManager/DataLakeStore/Commands.DataLakeStore/Commands/SetAzureRmDataLakeStoreFirewallRule.cs

@@ -21,7 +21,7 @@
 
 namespace Microsoft.Azure.Commands.DataLakeStore
 {
-    [Cmdlet(VerbsCommon.Set, "AzureRmDataLakeStoreFirewallRule", SupportsShouldProcess = true), OutputType(typeof(FirewallRule))]
+    [Cmdlet(VerbsCommon.Set, "AzureRmDataLakeStoreFirewallRule", SupportsShouldProcess = true), OutputType(typeof(DataLakeStoreFirewallRule))]
     [Alias("Set-AdlStoreFirewallRule")]
     public class SetAzureRmDataLakeStoreFirewallRule : DataLakeStoreCmdletBase
     {
@@ -67,8 +67,8 @@ public override void ExecuteCmdlet()
                 string.Format(Resources.SetDataLakeFirewallRule, Name),
                 Name,
                 () => 
-                    WriteObject(DataLakeStoreClient.AddOrUpdateFirewallRule(
-                        ResourceGroupName, Account, Name, startIp, endIp))
+                    WriteObject(new DataLakeStoreFirewallRule(DataLakeStoreClient.AddOrUpdateFirewallRule(
+                        ResourceGroupName, Account, Name, startIp, endIp)))
             );
         }
     }

src/ResourceManager/DataLakeStore/Commands.DataLakeStore/Commands/SetAzureRmDataLakeStoreTrustedIdProvider.cs

@@ -21,7 +21,7 @@
 
 namespace Microsoft.Azure.Commands.DataLakeStore
 {
-    [Cmdlet(VerbsCommon.Set, "AzureRmDataLakeStoreTrustedIdProvider", SupportsShouldProcess = true), OutputType(typeof(TrustedIdProvider))]
+    [Cmdlet(VerbsCommon.Set, "AzureRmDataLakeStoreTrustedIdProvider", SupportsShouldProcess = true), OutputType(typeof(DataLakeStoreTrustedIdProvider))]
     [Alias("Set-AdlStoreTrustedIdProvider")]
     public class SetAzureRmDataLakeStoreTrustedIdProvider : DataLakeStoreCmdletBase
     {
@@ -53,8 +53,8 @@ public override void ExecuteCmdlet()
                 string.Format(Resources.SetDataLakeTrustedIdProvider, Name),
                 Name,
                 () =>
-                    WriteObject(DataLakeStoreClient.AddOrUpdateTrustedProvider(
-                        ResourceGroupName, Account, Name, ProviderEndpoint))
+                    WriteObject(new DataLakeStoreTrustedIdProvider(DataLakeStoreClient.AddOrUpdateTrustedProvider(
+                        ResourceGroupName, Account, Name, ProviderEndpoint)))
             );
         }
     }