diff --git a/encompass-cs/Engine.cs b/encompass-cs/Engine.cs
index b3e69e7..cca223e 100644
--- a/encompass-cs/Engine.cs
+++ b/encompass-cs/Engine.cs
@@ -293,7 +293,7 @@ namespace Encompass
             return SetComponent(GetEntityByComponentID<TComponent>(componentID), component);
         }
 
-        protected Guid SetDrawComponent<TComponent>(Entity entity, TComponent component, int layer = 0) where TComponent : struct, IComponent
+        protected Guid SetDrawComponent<TComponent>(Entity entity, TComponent component, int layer = 0) where TComponent : struct, IComponent, IDrawComponent
         {
             var priority = writePriorities.ContainsKey(typeof(TComponent)) ? writePriorities[typeof(TComponent)] : 0;
 
diff --git a/encompass-cs/WorldBuilder.cs b/encompass-cs/WorldBuilder.cs
index e8176b7..be5d854 100644
--- a/encompass-cs/WorldBuilder.cs
+++ b/encompass-cs/WorldBuilder.cs
@@ -55,7 +55,7 @@ namespace Encompass
             return componentManager.MarkComponentForWrite(entity, component, priority);
         }
 
-        public Guid SetDrawComponent<TComponent>(Entity entity, TComponent component, int priority = 0, int layer = 0) where TComponent : struct, IComponent
+        public Guid SetDrawComponent<TComponent>(Entity entity, TComponent component, int priority = 0, int layer = 0) where TComponent : struct, IComponent, IDrawComponent
         {
             return componentManager.MarkDrawComponentForWrite(entity, component, priority, layer);
         }