[MLX4] cleanup.