Update stat registration to be like KV stats