From f86ff8bf1ab474eca4990af08c3ac751bad69da7 Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Wed, 14 Mar 2018 07:32:11 -0500
Subject: [PATCH] Set root universe in find_cell if not set

---
 src/geometry.F90 | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/geometry.F90 b/src/geometry.F90
index 10474848f..6096563d8 100644
--- a/src/geometry.F90
+++ b/src/geometry.F90
@@ -206,9 +206,12 @@ contains
     end do
     j = p % n_coord
 
-    ! Determine universe (if not set, use root universe
+    ! Determine universe (if not yet set, use root universe)
     i_universe = p % coord(j) % universe
-    if (i_universe == NONE) i_universe = root_universe
+    if (i_universe == NONE) then
+      p % coord(j) % universe = root_universe
+      i_universe = root_universe
+    end if
 
     ! set size of list to search
     if (present(search_cells)) then