From fb18d8a5d8b3dc466ec68f29a49fda8c4a03db8e Mon Sep 17 00:00:00 2001
From: Rowan Cockett <rowan@3ptscience.com>
Date: Sat, 7 Nov 2015 18:19:40 -0800
Subject: [PATCH] Dirty edges

---
 SimPEG/Mesh/PointerTree.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/SimPEG/Mesh/PointerTree.py b/SimPEG/Mesh/PointerTree.py
index 73855e27..d7cc1706 100644
--- a/SimPEG/Mesh/PointerTree.py
+++ b/SimPEG/Mesh/PointerTree.py
@@ -656,7 +656,9 @@ class Tree(object):
         self.__dirtyFaces__ = False
 
     def _numberEdges(self, force=False):
-        if self.dim == 2: return
+        if self.dim == 2:
+            self.__dirtyEdges__ = False
+            return
         if not self.__dirtyEdges__ and not force: return
 
         self._edgesX = set()
@@ -913,8 +915,8 @@ class Tree(object):
 
     def number(self, force=False):
         if not self.__dirty__ and not force: return
+        self.balance()
         self._hanging(force=force)
-        return
 
     def _deflationMatrix(self, theSet, theHang, theIndex, withHanging=True):
         reducedInd = dict() # final reduced index