Fix remove screen button.

diff --git a/src/wx/kdm_dialog.cc b/src/wx/kdm_dialog.cc
index 295287f42a13f323478655f8ac33034a76b25000..0336d0cb7f5484d1244c286509745e43041915c0 100644 (file)
--- 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()) {