Add in additional assert for an invalid axis