{
        hba->pwr_info.gear_rx = UFS_PWM_G1;
        hba->pwr_info.gear_tx = UFS_PWM_G1;
-       hba->pwr_info.lane_rx = 1;
-       hba->pwr_info.lane_tx = 1;
+       hba->pwr_info.lane_rx = UFS_LANE_1;
+       hba->pwr_info.lane_tx = UFS_LANE_1;
        hba->pwr_info.pwr_rx = SLOWAUTO_MODE;
        hba->pwr_info.pwr_tx = SLOWAUTO_MODE;
        hba->pwr_info.hs_rate = 0;
 
 void ufshcd_init_pwr_dev_param(struct ufs_dev_params *dev_param)
 {
        *dev_param = (struct ufs_dev_params){
-               .tx_lanes = 2,
-               .rx_lanes = 2,
+               .tx_lanes = UFS_LANE_2,
+               .rx_lanes = UFS_LANE_2,
                .hs_rx_gear = UFS_HS_G3,
                .hs_tx_gear = UFS_HS_G3,
                .pwm_rx_gear = UFS_PWM_G4,
 
        UFS_HS_G5               /* HS Gear 5 */
 };
 
+enum ufs_lanes {
+       UFS_LANE_DONT_CHANGE,   /* Don't change Lane */
+       UFS_LANE_1,             /* Lane 1 (default for reset) */
+       UFS_LANE_2,             /* Lane 2 */
+};
+
 enum ufs_unipro_ver {
        UFS_UNIPRO_VER_RESERVED = 0,
        UFS_UNIPRO_VER_1_40 = 1, /* UniPro version 1.40 */