diff --git a/util/Util.js b/util/Util.js
index 5e0c06f9..e3b254fd 100644
--- a/util/Util.js
+++ b/util/Util.js
@@ -219,7 +219,7 @@ module.exports = class Util {
 
 	static async isImageNSFW(model, image, bool = true) {
 		const img = await tf.node.decodeImage(image, 3);
-		const predictions = await model.classify(img, 2);
+		const predictions = await model.classify(img);
 		img.dispose();
 		if (bool) {
 			const results = [];