diff --git a/encompass-cs/WorldBuilder.cs b/encompass-cs/WorldBuilder.cs
index 7da9ced..2fe307e 100644
--- a/encompass-cs/WorldBuilder.cs
+++ b/encompass-cs/WorldBuilder.cs
@@ -82,7 +82,7 @@ namespace Encompass
///
/// Sets Component data for the specified Component Type on the specified Entity.
///
- public void SetComponent(Entity entity, in TComponent component) where TComponent : struct
+ public void SetComponent(Entity entity, in TComponent component) where TComponent : struct, IComponent
{
RegisterComponentType();
_startingExistingComponentStore.Set(entity.ID, component);
@@ -95,7 +95,7 @@ namespace Encompass
}
}
- internal void RegisterComponentType() where TComponent : struct
+ internal void RegisterComponentType() where TComponent : struct, IComponent
{
if (!_typeToIndex.ContainsKey(typeof(TComponent)))
{