From 0826aa4bfe8628265dadde5cf27d1f7e668b1ae2 Mon Sep 17 00:00:00 2001
From: blitzmann <ryan.xgamer99@gmail.com>
Date: Mon, 24 Feb 2014 21:58:42 -0500
Subject: [PATCH] Fix for issue 52

---
 gui/builtinViews/fittingView.py | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/gui/builtinViews/fittingView.py b/gui/builtinViews/fittingView.py
index f079adbfa..21bd543a6 100644
--- a/gui/builtinViews/fittingView.py
+++ b/gui/builtinViews/fittingView.py
@@ -230,7 +230,7 @@ class FittingView(d.Display):
                 self.removeModule(self.mods[row])
                 self.Select(row,0)
                 row = self.GetNextSelected(row)
-               
+
         event.Skip()
 
     def fitRemoved(self, event):
@@ -288,9 +288,8 @@ class FittingView(d.Display):
                     wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))
                 else:
                     populate = cFit.appendModule(fitID, itemID)
-                    if populate:
-                        self.slotsChanged()
                     if populate is not None:
+                        self.slotsChanged()
                         wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))
 
         event.Skip()
@@ -304,14 +303,14 @@ class FittingView(d.Display):
             else:
                 if "wxMSW" in wx.PlatformInfo:
                     self.click(event)
-                    
+
     def removeModule(self, module):
         cFit = service.Fit.getInstance()
         fit = cFit.getFit(self.activeFitID)
         populate = cFit.removeModule(self.activeFitID, fit.modules.index(module))
-        
+
         if populate is not None:
-            if populate: self.slotsChanged()
+            self.slotsChanged()
             wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.activeFitID))
 
     def swapItems(self, x, y, itemID):
@@ -373,16 +372,16 @@ class FittingView(d.Display):
             if self.activeFitID is not None and self.activeFitID == event.fitID:
                 self.generateMods()
                 self.refresh(self.mods)
-                
+
                 exportHtml.getInstance().refreshFittingHTMl()
-                
+
             self.Show(self.activeFitID is not None and self.activeFitID == event.fitID)
         except wx._core.PyDeadObjectError:
             pass
         finally:
             event.Skip()
 
-            
+
 
     def scheduleMenu(self, event):
         event.Skip()
@@ -451,7 +450,7 @@ class FittingView(d.Display):
                      5: ''}
     def slotColour(self, slot):
         return self.slotColourMap[slot] or self.GetBackgroundColour()
-    
+
     def refresh(self, stuff):
         d.Display.refresh(self, stuff)
         sFit = service.Fit.getInstance()
@@ -460,7 +459,7 @@ class FittingView(d.Display):
         for slotType in Slot.getTypes():
             slot = Slot.getValue(slotType)
             slotMap[slot] = fit.getSlotsFree(slot) < 0
-            
+
         for i, mod in enumerate(self.mods):
             if slotMap[mod.slot]:
                 self.SetItemBackgroundColour(i, wx.Colour(204, 51, 51))