diff --git a/bob/ip/binseg/engine/predictor.py b/bob/ip/binseg/engine/predictor.py
index 0a63d74a4a9501a2e4a54118491c52d918d8d7a6..62446d1549b3da4193a6c81fba026ef99757b176 100644
--- a/bob/ip/binseg/engine/predictor.py
+++ b/bob/ip/binseg/engine/predictor.py
@@ -14,6 +14,8 @@ import torchvision.transforms.functional as VF
 
 import bob.io.base
 
+from ..utils.summary import summary
+
 import logging
 logger = logging.getLogger(__name__)