diff --git a/src/cli.js b/src/cli.js
index e4001e5..1c3d831 100755
--- a/src/cli.js
+++ b/src/cli.js
@@ -94,7 +94,7 @@ require('yargs')
           alias: 'max-depth',
           demandOption: false,
           describe: 'Max depth to represent',
-          type: 'boolean',
+          type: 'integer',
         });
     },
     async (argv) => {