Fix specs to pass with changes to axiom