Optimize extra matrix multiplication