Update for deprecation of At_mul_B