diff --git a/pytorch3dunet/unet3d/config.py b/pytorch3dunet/unet3d/config.py
index bb011632..0dbecffd 100644
--- a/pytorch3dunet/unet3d/config.py
+++ b/pytorch3dunet/unet3d/config.py
@@ -49,7 +49,7 @@ def load_config():
     if device == 'cpu':
         logger.warning('CPU mode forced in config, this will likely result in slow training/prediction')
         config['device'] = 'cpu'
-        return config
+        return config, config_path
 
     if torch.cuda.is_available():
         config['device'] = 'cuda'
diff --git a/pytorch3dunet/unet3d/seg_metrics.py b/pytorch3dunet/unet3d/seg_metrics.py
index e713ea23..ddafe59d 100644
--- a/pytorch3dunet/unet3d/seg_metrics.py
+++ b/pytorch3dunet/unet3d/seg_metrics.py
@@ -29,7 +29,7 @@ def _iou_matrix(gt, seg):
     seg = _relabel(seg)
 
     # get number of overlapping pixels between GT and SEG
-    n_inter = contingency_table(gt, seg).A
+    n_inter = contingency_table(gt, seg).toarray()
 
     # number of pixels for GT instances
     n_gt = n_inter.sum(axis=1, keepdims=True)