Fix a couple incorrect equals(axis) calls

Old code tested on AxisType rather than Axis<?>. Maintain this
contract.