OK, I was just asking because I haven't looked much at this code and wanted to make sure it's correct. I'll let @shivaram look at it too.