if (r)
                goto err;
 
-       src->ops->dsi.disable(src, false, true);
+       src->ops->dsi.ulps(src, true);
 
        ddata->ulps_enabled = true;
 
        if (!ddata->ulps_enabled)
                return 0;
 
-       src->ops->enable(src);
+       src->ops->dsi.ulps(src, false);
        ddata->dsi->mode_flags &= ~MIPI_DSI_MODE_LPM;
 
        r = _dsicm_enable_te(ddata, true);
 
        dsicm_hw_reset(ddata);
 
-       src->ops->dsi.disable(src, true, false);
+       src->ops->disable(src);
 err_regulators:
        r = regulator_bulk_disable(ARRAY_SIZE(ddata->supplies), ddata->supplies);
        if (r)
                dsicm_hw_reset(ddata);
        }
 
-       src->ops->dsi.disable(src, true, false);
+       src->ops->disable(src);
 
        r = regulator_bulk_disable(ARRAY_SIZE(ddata->supplies), ddata->supplies);
        if (r)
 
        }
 }
 
-static void dsi_display_enable(struct omap_dss_device *dssdev)
+static void _dsi_display_enable(struct dsi_data *dsi)
 {
-       struct dsi_data *dsi = to_dsi_data(dssdev);
        int r;
 
-       DSSDBG("dsi_display_enable\n");
-
-       dsi_bus_lock(dsi);
+       WARN_ON(!dsi_bus_is_locked(dsi));
 
        mutex_lock(&dsi->lock);
 
 
        mutex_unlock(&dsi->lock);
 
-       dsi_bus_unlock(dsi);
-
        return;
 
 err_init_dsi:
        dsi_runtime_put(dsi);
 err_get_dsi:
        mutex_unlock(&dsi->lock);
-       DSSDBG("dsi_display_enable FAILED\n");
+       DSSDBG("dsi_display_ulps_enable FAILED\n");
 }
 
-static void dsi_display_disable(struct omap_dss_device *dssdev,
-               bool disconnect_lanes, bool enter_ulps)
+static void dsi_display_enable(struct omap_dss_device *dssdev)
 {
        struct dsi_data *dsi = to_dsi_data(dssdev);
-
-       DSSDBG("dsi_display_disable\n");
+       DSSDBG("dsi_display_enable\n");
 
        dsi_bus_lock(dsi);
+       _dsi_display_enable(dsi);
+       dsi_bus_unlock(dsi);
+}
+
+static void _dsi_display_disable(struct dsi_data *dsi,
+               bool disconnect_lanes, bool enter_ulps)
+{
+       WARN_ON(!dsi_bus_is_locked(dsi));
 
        mutex_lock(&dsi->lock);
 
        dsi_runtime_put(dsi);
 
        mutex_unlock(&dsi->lock);
+}
+
+static void dsi_display_disable(struct omap_dss_device *dssdev)
+{
+       struct dsi_data *dsi = to_dsi_data(dssdev);
+
+       DSSDBG("dsi_display_disable\n");
+
+       dsi_bus_lock(dsi);
+       _dsi_display_disable(dsi, true, false);
+       dsi_bus_unlock(dsi);
+}
+
+static void dsi_ulps(struct omap_dss_device *dssdev, bool enable)
+{
+       struct dsi_data *dsi = to_dsi_data(dssdev);
+
+       DSSDBG("dsi_ulps %d\n", enable);
+
+       dsi_bus_lock(dsi);
+
+       if (enable)
+               _dsi_display_disable(dsi, false, true);
+       else
+               _dsi_display_enable(dsi);
 
        dsi_bus_unlock(dsi);
 }
        .connect = dsi_connect,
        .disconnect = dsi_disconnect,
        .enable = dsi_display_enable,
+       .disable = dsi_display_disable,
 
        .dsi = {
-               .disable = dsi_display_disable,
+               .ulps = dsi_ulps,
 
                .set_config = dsi_set_config,