diff --git a/src/neuron_proofreader/merge_proofreading/merge_dataloading.py b/src/neuron_proofreader/merge_proofreading/merge_dataloading.py
index eb54c20..0bf8656 100644
--- a/src/neuron_proofreader/merge_proofreading/merge_dataloading.py
+++ b/src/neuron_proofreader/merge_proofreading/merge_dataloading.py
@@ -15,7 +15,7 @@
 
 from neuron_proofreader.utils import util
 
-TEST_BRAIN = "653159"
+TEST_BRAIN = "747807"
 
 
 # --- Load Skeletons ---
diff --git a/src/neuron_proofreader/split_proofreading/split_inference.py b/src/neuron_proofreader/split_proofreading/split_inference.py
index 9bcdf17..ec205fd 100644
--- a/src/neuron_proofreader/split_proofreading/split_inference.py
+++ b/src/neuron_proofreader/split_proofreading/split_inference.py
@@ -57,6 +57,7 @@ def __init__(
         img_path,
         model_path,
         output_dir,
+        model,
         config,
         log_preamble="",
         segmentation_path=None,
@@ -92,7 +93,7 @@ def __init__(
         self.accepted_proposals = list()
         self.config = config
         self.img_path = img_path
-        self.model = VisionHGAT(config.ml.patch_shape)
+        self.model = model
         self.output_dir = output_dir
         self.soma_centroids = soma_centroids