diff --git a/spkl/SparkleXrm.Tasks.Tests/ImportCodeTests.cs b/spkl/SparkleXrm.Tasks.Tests/ImportCodeTests.cs
index b9a625a..57c75fc 100644
--- a/spkl/SparkleXrm.Tasks.Tests/ImportCodeTests.cs
+++ b/spkl/SparkleXrm.Tasks.Tests/ImportCodeTests.cs
@@ -142,7 +142,7 @@ public void RegisterPluginCode()
 
             #region Assert
             Assert.AreEqual(1, created.Where(a => a.GetType() == typeof(PluginAssembly)).Count(), "1 Assembly");
-            Assert.AreEqual(1, created.Where(a => a.GetType() == typeof(PluginType)).Count(), "1 Type");
+            Assert.AreEqual(2, created.Where(a => a.GetType() == typeof(PluginType)).Count(), "2 Types (plugin & custom API)");
             Assert.AreEqual(4, created.Where(a => a.GetType() == typeof(SdkMessageProcessingStep)).Count(), "4 Steps");
             var step1 = created.Where(a => a.GetType() == typeof(SdkMessageProcessingStep)).FirstOrDefault().ToEntity<SdkMessageProcessingStep>();
             Assert.AreEqual(step1.Name, "Pre-Update of account", "Name check");