diff --git a/internal/ai/classify/model_test.go b/internal/ai/classify/model_test.go
index 8483480b8..cab503929 100644
--- a/internal/ai/classify/model_test.go
+++ b/internal/ai/classify/model_test.go
@@ -224,7 +224,7 @@ func TestModel_LoadModel(t *testing.T) {
 		err := tensorFlow.loadModel()
 
 		if err != nil {
-			assert.Contains(t, err.Error(), "no such file or directory")
+			assert.Contains(t, err.Error(), "Could not find SavedModel")
 		}
 
 		assert.Error(t, err)
diff --git a/internal/ai/nsfw/nsfw_test.go b/internal/ai/nsfw/nsfw_test.go
index 59e860b11..41959a050 100644
--- a/internal/ai/nsfw/nsfw_test.go
+++ b/internal/ai/nsfw/nsfw_test.go
@@ -13,7 +13,7 @@ import (
 
 var modelPath, _ = filepath.Abs("../../../assets/nsfw")
 
-var detector = NewModel(modelPath, 224, nil, false)
+var detector = NewModel(modelPath, nil, false)
 
 func TestIsSafe(t *testing.T) {
 	detect := func(filename string) Result {