free muon_energy struct in test_muon_get_energy()

1 files changed, 10 insertions, 2 deletions

diff --git a/src/test.c b/src/test.c
index 6bb2046..e46963b 100644
--- a/src/test.c
+++ b/src/test.c
@@ -125,7 +125,7 @@ int test_muon_get_energy(char *err)
     /* At the beginning of the track we should have roughly the same energy. */
     if (!isclose(T, T0, 1e-5, 0)) {
         sprintf(err, "KE = %.5f, but expected %.5f", T, T0);
-        return 1;
+        goto err;
     }
 
     range = get_range(T0,1.0);
@@ -134,10 +134,18 @@ int test_muon_get_energy(char *err)
     /* At the end of the track the energy should be approximately 0. */
     if (!isclose(T, 0, 1e-5, 1e-5)) {
         sprintf(err, "KE = %.5f, but expected %.5f", T, 0.0);
-        return 1;
+        goto err;
     }
 
+    muon_free_energy(m);
+
     return 0;
+
+err:
+    muon_free_energy(m);
+
+    return 1;
+
 }
 
 int test_muon_get_range(char *err)