Fix W matrix klda