CitiesSkylinesMods#744

I return after update to avoid exception.
kianzarrin · Feb 26, 2020 · 2414315 · 2414315
1 parent 13ca9b8
commit 2414315
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/TLM/TLM/UI/SubTools/VehicleRestrictionsTool.cs b/TLM/TLM/UI/SubTools/VehicleRestrictionsTool.cs
@@ -218,7 +218,6 @@ public override void ShowGUIOverlay(ToolMode toolMode, bool viewOnly) {
         private void ShowSigns(bool viewOnly) {
             Vector3 camPos = Camera.main.transform.position;
             NetManager netManager = Singleton<NetManager>.instance;
-            ushort updatedSegmentId = 0;
             bool handleHovered = false;
 
             foreach (ushort segmentId in currentRestrictedSegmentIds) {
@@ -248,15 +247,11 @@ private void ShowSigns(bool viewOnly) {
                 }
 
                 if (update) {
-                    updatedSegmentId = segmentId;
+                    break;
                 }
             }
 
             overlayHandleHovered = handleHovered;
-
-            if (updatedSegmentId != 0) {
-                RefreshCurrentRestrictedSegmentIds(updatedSegmentId);
-            }
         }
 
         private void GuiVehicleRestrictionsWindow(int num) {
@@ -581,7 +576,7 @@ private bool DrawVehicleRestrictionHandles(ushort segmentId,
                             vehicleType,
                             !allowed);
                         stateUpdated = true;
-
+                        RefreshCurrentRestrictedSegmentIds(segmentId);
                         if (RoadMode) {
                             ApplyRestrictionsToAllSegments(sortedLaneIndex, vehicleType);
                         }