diff --git a/cellbender/remove_background/run.py b/cellbender/remove_background/run.py
index 7ac466b..5e13e83 100644
--- a/cellbender/remove_background/run.py
+++ b/cellbender/remove_background/run.py
@@ -111,7 +111,7 @@ def run_remove_background(args: argparse.Namespace) -> Posterior:
     if args.model == 'naive':
         inferred_model = None
     else:
-        inferred_model, _, _, _ = run_inference(dataset_obj=dataset_obj, args=args)
+        inferred_model, _, _, _ = run_inference(dataset_obj=dataset_obj, args=args, output_checkpoint_tarball=args.input_checkpoint_tarball)
         inferred_model.eval()
 
     try: