Index: trunk/SDToolBox/extract_data_utils.py =================================================================== diff -u -r213 -r214 --- trunk/SDToolBox/extract_data_utils.py (.../extract_data_utils.py) (revision 213) +++ trunk/SDToolBox/extract_data_utils.py (.../extract_data_utils.py) (revision 214) @@ -103,7 +103,7 @@ """ unravel_idx = np.unravel_index(index, idx_shape) if not isinstance(unravel_idx[0], (list, np.ndarray)): - return np.array(unravel_idx[0]), np.array(unravel_idx[1]) + return np.array([unravel_idx[0]]), np.array([unravel_idx[1]]) return unravel_idx