diff --git a/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.csproj b/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.csproj
index f5a6182c..6c1554d4 100644
--- a/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.csproj
+++ b/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.csproj
@@ -3,7 +3,7 @@
   <PropertyGroup>
     <OutputType>Exe</OutputType>
     <TargetFramework>netcoreapp2.1</TargetFramework>    
-    <Version>2.2.0</Version>
+    <Version>2.2.1</Version>
     <AssemblyName>FunctionMonkey.Compiler</AssemblyName>
     <PackageId>FunctionMonkey.Compiler</PackageId>
   </PropertyGroup>
@@ -385,7 +385,7 @@
     <IntermediatePackDir>$(MSBuildProjectDirectory)/bin/$(Configuration)/publish/</IntermediatePackDir>
     <PublishDir>$(IntermediatePackDir)$(TargetFramework)/</PublishDir>
     <NuspecProperties>publishDir=$([MSBuild]::NormalizeDirectory($(IntermediatePackDir)))</NuspecProperties>
-    <AssemblyVersion>2.2.0.0</AssemblyVersion>
+    <AssemblyVersion>2.2.1.0</AssemblyVersion>
     <FileVersion>2.2.1.0</FileVersion>
     <PackageReleaseNotes></PackageReleaseNotes>
   </PropertyGroup>
diff --git a/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.nuspec b/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.nuspec
index e18a0a38..fd6a69cf 100644
--- a/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.nuspec
+++ b/Source/FunctionMonkey.Compiler/FunctionMonkey.Compiler.nuspec
@@ -2,7 +2,7 @@
 <package xmlns="http://schemas.microsoft.com/packaging/2012/06/nuspec.xsd">
   <metadata>
     <id>FunctionMonkey.Compiler</id>
-    <version>2.2.0</version>
+    <version>2.2.1</version>
     <authors>James Randall</authors>
     <description>Generates Azure Functions from command registrations</description>
     <licenseUrl>https://raw.githubusercontent.com/JamesRandall/AzureFromTheTrenches.Commanding/master/LICENSE</licenseUrl>
diff --git a/Source/FunctionMonkey.Compiler/Implementation/AssemblyCompiler.cs b/Source/FunctionMonkey.Compiler/Implementation/AssemblyCompiler.cs
index fef88c37..14eb0cf2 100644
--- a/Source/FunctionMonkey.Compiler/Implementation/AssemblyCompiler.cs
+++ b/Source/FunctionMonkey.Compiler/Implementation/AssemblyCompiler.cs
@@ -272,6 +272,7 @@ private static IReadOnlyCollection<string> BuildCandidateReferenceList(IReadOnly
                 typeof(Document).GetTypeInfo().Assembly.Location,
                 typeof(Message).GetTypeInfo().Assembly.Location,
                 typeof(ChangeFeedProcessorBuilder).Assembly.Location,
+                typeof(CosmosDBAttribute).Assembly.Location,
                 typeof(TimerInfo).Assembly.Location,
                 typeof(DbConnectionStringBuilder).Assembly.Location,
                 typeof(AzureSignalRAuthClient).Assembly.Location,