From 7c470a47b04b9dd3ec072fd815ad885329bccbcf Mon Sep 17 00:00:00 2001
From: bastiencarreres <bastien.carreres@duke.edu>
Date: Wed, 28 Aug 2024 16:44:56 -0400
Subject: [PATCH] up tests

---
 snsim/tests/test_astrobj.py | 1 +
 snsim/tests/test_survey.py  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/snsim/tests/test_astrobj.py b/snsim/tests/test_astrobj.py
index 2dc488e..5245140 100644
--- a/snsim/tests/test_astrobj.py
+++ b/snsim/tests/test_astrobj.py
@@ -35,6 +35,7 @@ def setup_class(self):
             "beta": 3.1,
             "model_name": "salt2",
             "model_version": "2.4",
+            "mass_step": 0.0
         }
 
         self.SNIa_Tripp = sn_astrobj.SNIa(sim_par, mag_fun="SALTTripp")
diff --git a/snsim/tests/test_survey.py b/snsim/tests/test_survey.py
index 9220507..98e3c29 100644
--- a/snsim/tests/test_survey.py
+++ b/snsim/tests/test_survey.py
@@ -1,2 +1,2 @@
 import snsim
-from snsim.survey_host import SurveyObs
+from snsim.survey import SurveyObs