change PWA d_dx to return a dead axis like affine