Yes, you should be able to with that commit. I've added a test to make sure this works for both edition and creation.