I'm not entirely sure about this line -- the tests crash here for axis=1,2, complaining about a loss of precision