From 666825da92cba00cd70efbeb2a032d5dfe79caaf Mon Sep 17 00:00:00 2001
From: Carl Hetherington <cth@carlh.net>
Date: Mon, 31 Mar 2014 14:11:57 +0100
Subject: [PATCH] Fix remove screen button.

---
 src/wx/kdm_dialog.cc | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/wx/kdm_dialog.cc b/src/wx/kdm_dialog.cc
index 295287f42..0336d0cb7 100644
--- a/src/wx/kdm_dialog.cc
+++ b/src/wx/kdm_dialog.cc
@@ -354,9 +354,11 @@ KDMDialog::remove_screen_clicked ()
 	pair<wxTreeItemId, shared_ptr<Screen> > s = selected_screens().front();
 
 	map<wxTreeItemId, shared_ptr<Cinema> >::iterator i = _cinemas.begin ();
-	list<shared_ptr<Screen> > sc = i->second->screens ();
-	while (i != _cinemas.end() && find (sc.begin(), sc.end(), s.second) == sc.end()) {
-		++i;
+	while (i != _cinemas.end ()) {
+		list<shared_ptr<Screen> > sc = i->second->screens ();
+		if (find (sc.begin(), sc.end(), s.second) != sc.end ()) {
+			break;
+		}
 	}
 
 	if (i == _cinemas.end()) {
-- 
2.30.2