diff --git a/Sources/FirebladeECS/FSM.swift b/Sources/FirebladeECS/FSM.swift
index 72c3c35..d99ac8d 100644
--- a/Sources/FirebladeECS/FSM.swift
+++ b/Sources/FirebladeECS/FSM.swift
@@ -210,6 +210,7 @@ public class StateComponentMapping {
         self.creatingState = creatingState
         componentType = type
         provider = ComponentTypeProvider(type: type)
+        creatingState.providers[componentType.identifier] = provider
     }
 
     /// Creates a mapping for the component type to a specific component instance. A