diff --git a/bob/learn/tensorflow/extractors/__init__.py b/bob/learn/tensorflow/extractors/__init__.py
index ac13aeec4364c6d2481e00aa2d0c37abfb94407e..05492a8e524ac60164f5be396ab9597ddcc2ec88 100644
--- a/bob/learn/tensorflow/extractors/__init__.py
+++ b/bob/learn/tensorflow/extractors/__init__.py
@@ -1,4 +1,4 @@
-from .Base import Base
+from .Base import Base, normalize_checkpoint_path
 from .Generic import Generic
 from .Estimator import Estimator