struct regmap *regmap, int irq)
 {
        int i, ret, fmin, fmax, freq, uartclk;
-       struct clk *clk_osc, *clk_xtal;
        struct max310x_port *s;
        bool xtal = false;
 
                return -ENOMEM;
        }
 
-       clk_osc = devm_clk_get(dev, "osc");
-       clk_xtal = devm_clk_get(dev, "xtal");
-       if (!IS_ERR(clk_osc)) {
-               s->clk = clk_osc;
+       s->clk = devm_clk_get_optional(dev, "osc");
+       if (IS_ERR(s->clk))
+               return PTR_ERR(s->clk);
+       if (s->clk) {
                fmin = 500000;
                fmax = 35000000;
-       } else if (!IS_ERR(clk_xtal)) {
-               s->clk = clk_xtal;
-               fmin = 1000000;
-               fmax = 4000000;
-               xtal = true;
-       } else if (PTR_ERR(clk_osc) == -EPROBE_DEFER ||
-                  PTR_ERR(clk_xtal) == -EPROBE_DEFER) {
-               return -EPROBE_DEFER;
        } else {
-               dev_err(dev, "Cannot get clock\n");
-               return -EINVAL;
+               s->clk = devm_clk_get_optional(dev, "xtal");
+               if (IS_ERR(s->clk))
+                       return PTR_ERR(s->clk);
+               if (s->clk) {
+                       fmin = 1000000;
+                       fmax = 4000000;
+                       xtal = true;
+               } else {
+                       dev_err(dev, "Cannot get clock\n");
+                       return -EINVAL;
+               }
        }
 
        ret = clk_prepare_enable(s->clk);