From 8a275bbd77688bacdd1e16700429b598f28504d3 Mon Sep 17 00:00:00 2001
From: tjlaboss <travis@ktc-technology.com>
Date: Wed, 29 Mar 2017 20:12:24 -0400
Subject: [PATCH] Use Geometry.get_all_cells()

Geometry.get_all_cells() accounts for nested cells.
Geometry.root_universe.cells() doesn't.
---
 openmc/geometry.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/openmc/geometry.py b/openmc/geometry.py
index 8ef8161da..4e6142204 100644
--- a/openmc/geometry.py
+++ b/openmc/geometry.py
@@ -270,10 +270,8 @@ class Geometry(object):
         """
         surfaces = OrderedDict()
         
-        root_cells = self._root_universe.cells
-        for cell in root_cells.values():
-            reg = cell.region
-            surfaces = reg.get_surfaces_from_region(surfaces)
+        for cell in self.get_all_cells().values():
+            surfaces = cell.region.get_surfaces_from_region(surfaces)
         return surfaces
                 
     def get_materials_by_name(self, name, case_sensitive=False, matching=False):