Fix errors in AxesTests