updated build to support axiom-demo