[MLX4] fix in error flow.