try with a dimshuffle