Fix bi checks in Production lua.

MustaphaTR · Aug 23, 2023 · 6a9ac4d · 6a9ac4d
1 parent 118a095
commit 6a9ac4d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/OpenRA.Mods.Common/Scripting/Properties/ProductionProperties.cs b/OpenRA.Mods.Common/Scripting/Properties/ProductionProperties.cs
@@ -194,12 +194,13 @@ public bool IsProducing(string actorType)
 		BuildableInfo GetBuildableInfo(string actorType, string queue)
 		{
 			var ri = Self.World.Map.Rules.Actors[actorType];
-			var bi = BuildableInfo.GetTraitForQueue(ri, queue);
+			var bi = ri.TraitInfos<BuildableInfo>().FirstOrDefault();
 
+			// Error if there are no Buildable traits, return correct one otherwise.
 			if (bi == null)
 				throw new LuaException($"Actor of type {actorType} cannot be produced");
 			else
-				return bi;
+				return BuildableInfo.GetTraitForQueue(ri, queue);
 		}
 	}