Revert "Update Axion plugin"

This reverts commit 862c05e3c25ae399efc489c751e5ce9001dc704b.