fix argument ax to wigner_fock_distribution: ax must be a list of axes