Fix researching surveillance-2 not immediately unlocking extra survei…

…llance
Afforess · May 11, 2016 · 265a8b2 · 265a8b2
1 parent eb89cf3
commit 265a8b2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Makefile b/Makefile
@@ -1,5 +1,5 @@
 PACKAGE_NAME := Big_Brother
-VERSION_STRING := 0.2.2
+VERSION_STRING := 0.2.3
 
 OUTPUT_NAME := $(PACKAGE_NAME)_$(VERSION_STRING)
 OUTPUT_DIR := build/$(OUTPUT_NAME)

diff --git a/control.lua b/control.lua
@@ -88,17 +88,19 @@ end)
 
 Event.register(defines.events.on_research_finished, function(event)
     local tech_name = event.research.name
+    local force = event.research.force
     if tech_name:starts_with('radar-amplifier') or tech_name:starts_with('radar-efficiency') then
         -- update radars in 1 tick
-        local force = event.research.force
         Event.register(defines.events.on_tick, function(event)
             upgrade_radars(force)
-
             Event.remove(defines.events.on_tick, event._handler)
         end)
     elseif tech_name == 'surveillance-2' then
         if global.power_poles and global.surveillance_centers then
-            update_all_surveillance(event.research.force)
+            Event.register(defines.events.on_tick, function(event)
+                update_all_surveillance(force)
+                Event.remove(defines.events.on_tick, event._handler)
+            end)
         end
     end
 end)