A matrix ansatz for the diffusion of an impurity in the asymmetric exclusion process