Fix ecs behaviour

hadashiA · May 3, 2021 · 3400bb7 · 3400bb7
1 parent fcf65d3
commit 3400bb7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/VContainer/Assets/VContainer/Runtime/Unity/ContainerBuilderUnityExtensions.cs b/VContainer/Assets/VContainer/Runtime/Unity/ContainerBuilderUnityExtensions.cs
@@ -208,9 +208,8 @@ public static RegistrationBuilder RegisterSystemFromWorld<T>(this IContainerBuil
             if (system is null)
                 throw new ArgumentException($"{typeof(T).FullName} is not in the world {world}");
 
-            return builder.RegisterInstance(system)
-                .As(typeof(ComponentSystemBase))
-                .AsSelf();
+            return builder.RegisterComponent(system)
+                .As(typeof(ComponentSystemBase), typeof(T));
         }
 
         // Use custom world