(svn r23665) -Fix (r23664): revert, and apply other fix, which allows switching companies safely in SinglePlayer

1 files changed, 4 insertions, 3 deletions

diff --git a/src/goal_gui.cpp b/src/goal_gui.cpp
index aeba8e576..26d5238df 100644
--- a/src/goal_gui.cpp
+++ b/src/goal_gui.cpp
@@ -20,6 +20,7 @@
 #include "gui.h"
 #include "goal_base.h"
 #include "core/geometry_func.hpp"
+#include "company_func.h"
 
 #include "widgets/goal_widget.h"
 
@@ -63,7 +64,7 @@ struct GoalListWindow : Window {
 		if (y < 0) return;
 
 		FOR_ALL_GOALS(s) {
-			if (s->company != INVALID_COMPANY) {
+			if (s->company == _local_company) {
 				y--;
 				if (y == 0) {
 					this->HandleClick(s);
@@ -119,7 +120,7 @@ struct GoalListWindow : Window {
 		FOR_ALL_GOALS(s) {
 			if (s->company == INVALID_COMPANY) {
 				num_global++;
-			} else {
+			} else if (s->company == _local_company) {
 				num_company++;
 			}
 		}
@@ -189,7 +190,7 @@ struct GoalListWindow : Window {
 		num = 0;
 
 		FOR_ALL_GOALS(s) {
-			if (s->company != INVALID_COMPANY) {
+			if (s->company == _local_company) {
 				if (IsInsideMM(pos, 0, cap)) {
 					/* Display the goal */
 					SetDParamStr(0, s->text);