Update to match new primus api