Update to use new high dim api