From 7c5d3bfe2a624cc088f2700b7d8693825bddf957 Mon Sep 17 00:00:00 2001
From: Matthew McDermott <mattmcdermott8@gmail.com>
Date: Fri, 8 Nov 2024 11:32:44 -0500
Subject: [PATCH] Fixed small test typo

---
 tests/utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/utils.py b/tests/utils.py
index 71e60d6..4bb9741 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -405,6 +405,9 @@ def single_stage_tester(
     if df_check_kwargs is None:
         df_check_kwargs = {}
 
+    if stage_kwargs is None:
+        stage_kwargs = {}
+
     with input_dataset(input_files) as (input_dir, cohort_dir):
         for k, v in pipeline_kwargs.items():
             if type(v) is str and "{input_dir}" in v: