diff --git a/bob/ip/binseg/test/test_summary.py b/bob/ip/binseg/test/test_summary.py
index b29bfcc87af1568fe58d8df4984a01f3a74d1e00..512bb93f2bc5f0f5f599b5b6cefebe6eab663e75 100644
--- a/bob/ip/binseg/test/test_summary.py
+++ b/bob/ip/binseg/test/test_summary.py
@@ -4,11 +4,11 @@
 import os
 import unittest
 
-from ..modeling.driu import build_driu
-from ..modeling.driuod import build_driuod
-from ..modeling.hed import build_hed
-from ..modeling.unet import build_unet
-from ..modeling.resunet import build_res50unet
+from ..models.driu import driu
+from ..models.driu_od import driu_od
+from ..models.hed import hed
+from ..models.unet import unet
+from ..models.resunet import resunet50
 from ..utils.summary import summary
 
 
@@ -18,31 +18,31 @@ class Tester(unittest.TestCase):
     """
 
     def test_summary_driu(self):
-        model = build_driu()
+        model = driu()
         s, param = summary(model)
         self.assertIsInstance(s, str)
         self.assertIsInstance(param, int)
 
     def test_summary_driuod(self):
-        model = build_driuod()
+        model = driu_od()
         s, param = summary(model)
         self.assertIsInstance(s, str)
         self.assertIsInstance(param, int)
 
     def test_summary_hed(self):
-        model = build_hed()
+        model = hed()
         s, param = summary(model)
         self.assertIsInstance(s, str)
         self.assertIsInstance(param, int)
 
     def test_summary_unet(self):
-        model = build_unet()
+        model = unet()
         s, param = summary(model)
         self.assertIsInstance(s, str)
         self.assertIsInstance(param, int)
 
     def test_summary_resunet(self):
-        model = build_res50unet()
+        model = resunet()
         s, param = summary(model)
         self.assertIsInstance(s, str)
         self.assertIsInstance(param, int)