diff --git a/sbb_binarize/cli.py b/sbb_binarize/cli.py
index 0176e20..0077bef 100644
--- a/sbb_binarize/cli.py
+++ b/sbb_binarize/cli.py
@@ -8,7 +8,7 @@ from .sbb_binarize import SbbBinarizer
 @command()
 @version_option()
 @option('--patches/--no-patches', default=True, help='by enabling this parameter you let the model to see the image in patches.')
-@option('--model-dir', '-m', type=click.Path(exists=True, file_okay=False), required=True, help='directory containing models for prediction')
+@option('--model-dir', '-m', type=types.Path(exists=True, file_okay=False), required=True, help='directory containing models for prediction')
 @argument('input_image')
 @argument('output_image')
 def main(patches, model_dir, input_image, output_image):