Update to work with latest rig-builder.