From f215adb44f8b5e8cf52d2258917740063581c2e6 Mon Sep 17 00:00:00 2001
From: marshrossney <17361029+marshrossney@users.noreply.github.com>
Date: Mon, 26 Apr 2021 16:55:20 +0100
Subject: [PATCH] fix broken test due to class name change

---
 anvil/tests/test_distributions.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/anvil/tests/test_distributions.py b/anvil/tests/test_distributions.py
index 939e92a..b219bc9 100644
--- a/anvil/tests/test_distributions.py
+++ b/anvil/tests/test_distributions.py
@@ -5,7 +5,7 @@
 from math import sqrt
 
 import numpy as np
-from anvil.distributions import NormalDist
+from anvil.distributions import Gaussian
 
 MEAN = 0
 SIGMA = 1
@@ -20,7 +20,7 @@ def test_normal_distribution():
 
     """
     lattice_size = 5
-    generator = NormalDist(lattice_size, sigma=SIGMA, mean=MEAN)
+    generator = Gaussian(lattice_size, sigma=SIGMA, mean=MEAN)
     sample_pt, _ = generator(N_SAMPLE)
     sample_np = sample_pt.detach().numpy()
     np.testing.assert_allclose(