diff --git a/src/ServiceManagement/Services/Commands.Test/Profile/ProfileCmdltsTests.cs b/src/ServiceManagement/Services/Commands.Test/Profile/ProfileCmdltsTests.cs
index 705b9fde57ab..081523905fbe 100644
--- a/src/ServiceManagement/Services/Commands.Test/Profile/ProfileCmdltsTests.cs
+++ b/src/ServiceManagement/Services/Commands.Test/Profile/ProfileCmdltsTests.cs
@@ -23,6 +23,7 @@
 using Microsoft.WindowsAzure.Commands.Common;
 using Microsoft.WindowsAzure.Commands.Common.Test.Mocks;
 using Microsoft.WindowsAzure.Commands.Profile;
+using Microsoft.WindowsAzure.Commands.Profile.Models;
 using Microsoft.WindowsAzure.Commands.Utilities.Common;
 using System;
 using System.Collections.Generic;
@@ -731,7 +732,7 @@ public void SelectAzureSubscriptionWithPassthroughPrintsSubscription()
 
             // Verify
             Assert.Equal(1, commandRuntimeMock.OutputPipeline.Count);
-            Assert.True(commandRuntimeMock.OutputPipeline[0] is AzureSubscription);
+            Assert.True(commandRuntimeMock.OutputPipeline[0] is PSAzureSubscription);
         }
 
         [Fact]