diff --git a/doc/api.rst b/doc/api.rst
index a6608a637c5a55b0e73a3d15c54364c4b8f37ffa..1eade48273e9a1117484ac0d14a7df6b510b5e3d 100644
--- a/doc/api.rst
+++ b/doc/api.rst
@@ -17,9 +17,11 @@ PyTorch ImageFolder Dataset
 ===========================
 .. automodule:: bob.ip.binseg.data.imagefolder
 
+.. automodule:: bob.ip.binseg.data.imagefolderinference
+
 Transforms
 ==========
-.. note:: 
+.. note::
     All transforms work with :py:class:`PIL.Image.Image` objects. We make heavy use of the
     `torchvision package`_