diff --git a/tests/test_point_source.py b/tests/test_point_source.py
index 952719a..a8d7c28 100644
--- a/tests/test_point_source.py
+++ b/tests/test_point_source.py
@@ -14,7 +14,7 @@ def test_creation():
     # Ensure it has space, angle, and energy set
     assert isinstance(my_source.space, openmc.stats.Point)
     assert isinstance(my_source.angle, openmc.stats.Isotropic)
-    assert isinstance(my_source.energy, openmc.stats.muir)
+    assert isinstance(my_source.energy, openmc.stats.univariate.Normal)
 
 
 @pytest.mark.parametrize(
diff --git a/tests/test_ring_source.py b/tests/test_ring_source.py
index c5561a8..18c151e 100644
--- a/tests/test_ring_source.py
+++ b/tests/test_ring_source.py
@@ -14,7 +14,7 @@ def test_creation():
     # Ensure it has space, angle, and energy set
     assert isinstance(my_source.space, openmc.stats.CylindricalIndependent)
     assert isinstance(my_source.angle, openmc.stats.Isotropic)
-    assert isinstance(my_source.energy, openmc.stats.muir)
+    assert isinstance(my_source.energy, openmc.stats.univariate.Normal)
 
 
 @pytest.mark.parametrize("radius", [1.0, 5.6, 1e5, 7])