diff --git a/tests/test_union_energy_grids/results.py b/tests/test_union_energy_grids/results.py
index be13ee66f1..b616e2874b 100644
--- a/tests/test_union_energy_grids/results.py
+++ b/tests/test_union_energy_grids/results.py
@@ -2,20 +2,20 @@
 
 import sys
 
-# import statepoint
 sys.path.insert(0, '../../src/utils')
-import statepoint
+from openmc.statepoint import StatePoint
 
 # read in statepoint file
 if len(sys.argv) > 1:
-    sp = statepoint.StatePoint(sys.argv[1])
+    print(sys.argv)
+    sp = StatePoint(sys.argv[1])
 else:
-    sp = statepoint.StatePoint('statepoint.10.binary')
+    sp = StatePoint('statepoint.10.binary')
 sp.read_results()
 
 # set up output string
 outstr = ''
- 
+
 # write out k-combined
 outstr += 'k-combined:\n'
 outstr += "{0:12.6E} {1:12.6E}\n".format(sp.k_combined[0], sp.k_combined[1])