diff --git a/jecs.d.ts b/jecs.d.ts
index b577b67..b80cfc9 100644
--- a/jecs.d.ts
+++ b/jecs.d.ts
@@ -149,7 +149,7 @@ export class World {
 	 * @param component Target Component
 	 * @param value Component Value
 	 */
-	set(entity: Id, component: Entity, value: InferComponent<typeof component>): void;
+	set<TComponent extends Entity>(entity: Id, component: TComponent, value: InferComponent<NoInfer<TComponent>>): void;
 
 	/**
 	 * Removes a component from the given entity