static int cqspi_setup_flash(struct cqspi_st *cqspi)
 {
-       unsigned int max_cs = cqspi->num_chipselect - 1;
        struct platform_device *pdev = cqspi->pdev;
        struct device *dev = &pdev->dev;
        struct cqspi_flash_pdata *f_pdata;
-       unsigned int cs;
-       int ret;
+       int ret, cs, max_cs = -1;
 
        /* Get flash device data */
        for_each_available_child_of_node_scoped(dev->of_node, np) {
                if (cs >= cqspi->num_chipselect) {
                        dev_err(dev, "Chip select %d out of range.\n", cs);
                        return -EINVAL;
-               } else if (cs < max_cs) {
-                       max_cs = cs;
                }
 
+               max_cs = max_t(int, cs, max_cs);
+
                f_pdata = &cqspi->f_pdata[cs];
                f_pdata->cqspi = cqspi;
                f_pdata->cs = cs;
                        return ret;
        }
 
+       if (max_cs < 0) {
+               dev_err(dev, "No flash device declared\n");
+               return -ENODEV;
+       }
+
        cqspi->num_chipselect = max_cs + 1;
        return 0;
 }