diff --git a/metal_small/generate_files_for_tests.py b/metal_small/generate_files_for_tests.py
index d5c896e..48979ee 100644
--- a/metal_small/generate_files_for_tests.py
+++ b/metal_small/generate_files_for_tests.py
@@ -2,7 +2,6 @@
 import copy
 import stat
 import glob
-import numpy as np
 import argparse
 import asdf
 
@@ -79,7 +78,6 @@ def generate_files_for_tests(run_beast=True, run_tools=True):
                 use_sd=False,
                 nsubs=settings.n_subgrid,
                 nprocs=1,
-                use_rate=True,
             )
 
             # -----------------
diff --git a/metal_small/run_beast.py b/metal_small/run_beast.py
index 4f1727b..3a87648 100755
--- a/metal_small/run_beast.py
+++ b/metal_small/run_beast.py
@@ -69,7 +69,7 @@
         print("Generating noise model from ASTs and absflux A matrix")
 
         create_obsmodel.create_obsmodel(
-            settings, use_sd=False, nsubs=settings.n_subgrid, nprocs=1, use_rate=True,
+            settings, use_sd=False, nsubs=settings.n_subgrid, nprocs=1,
         )
 
         # in the absence of ASTs, the splinter noise model can be used