[MLX4] fix to the previous patch. [mlnx: 4417]