diff --git a/lib/collections.js b/lib/collections.js
index 195bf8e..de6f814 100644
--- a/lib/collections.js
+++ b/lib/collections.js
@@ -12,6 +12,7 @@ async function getCollection(collection_slug) {
 
 /**
  * Fetch a list of model collections
+ *
  * @returns {Promise<object>} - Resolves with the collections data
  */
 async function listCollections() {
diff --git a/lib/predictions.js b/lib/predictions.js
index 47f2b51..3ec8546 100644
--- a/lib/predictions.js
+++ b/lib/predictions.js
@@ -4,7 +4,7 @@
  * @param {object} options
  * @param {string} options.version - Required. The model version
  * @param {object} options.input - Required. An object with the model inputs
- * @param {object} [options.wait] - Whether to wait for the prediction to finish. Defaults to false
+ * @param {boolean|object} [options.wait] - Whether to wait for the prediction to finish. Defaults to false
  * @param {number} [options.wait.interval] - Polling interval in milliseconds. Defaults to 250
  * @param {number} [options.wait.maxAttempts] - Maximum number of polling attempts. Defaults to no limit
  * @param {string} [options.webhook] - An HTTPS URL for receiving a webhook when the prediction has new output