This does in fact work. I've updated the code locally and updated the tests with this. Will push once I figure out the Py3 failure.