allow axis type to be None