From 7a20de725df55fe0809c0a36cc0b6074bfeba5ef Mon Sep 17 00:00:00 2001
From: Teresa Gomez <46339554+teresamg@users.noreply.github.com>
Date: Fri, 17 May 2024 10:44:45 -0400
Subject: [PATCH] Updates get_segments()

---
 nireports/reportlets/modality/dwi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nireports/reportlets/modality/dwi.py b/nireports/reportlets/modality/dwi.py
index 63258db5..84465e82 100644
--- a/nireports/reportlets/modality/dwi.py
+++ b/nireports/reportlets/modality/dwi.py
@@ -524,7 +524,7 @@ def get_segments(
         indices = np.array([], dtype=int)
         for ii in idx:
             indices = np.concatenate(
-                [indices, np.nonzero(segment_mask == ii)]
+                [indices, np.nonzero(segment_mask == ii)[0]]
             )
         segments[label] = indices