diff --git a/imaginairy/api.py b/imaginairy/api.py
index 0aaafa8..e94eb30 100755
--- a/imaginairy/api.py
+++ b/imaginairy/api.py
@@ -253,8 +253,7 @@ def imagine(
                     upscaled_img = None
                     is_nsfw_img = None
                     if IMAGINAIRY_SAFETY_MODE != SafetyMode.DISABLED:
-                        if is_nsfw(img, x_sample, half_mode=half_mode):
-                            is_nsfw_img = True
+                        is_nsfw_img = is_nsfw(img, x_sample, half_mode=half_mode)
                         if is_nsfw_img and IMAGINAIRY_SAFETY_MODE == SafetyMode.FILTER:
                             logger.info("    ⚠️  Filtering NSFW image")
                             img = img.filter(ImageFilter.GaussianBlur(radius=40))