diff --git a/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c b/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
index 6bd9dd286057..7e9f518001e8 100644
--- a/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
+++ b/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
@@ -500,11 +500,12 @@ static int ion_sgl_sync_range(struct device *dev, struct scatterlist *sgl,
 			break;
 
 		if (i > 0) {
-			if (get_dma_ops(dev)) {
-				pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
-					"is dma_set_max_seg_size being set on dev:%s\n",
-					dev_name(dev));
-			}
+			if (!get_dma_ops(dev))
+				return 0;
+
+			pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
+				"is dma_set_max_seg_size being set on dev:%s\n",
+				dev_name(dev));
 			return -EINVAL;
 		}
 	}