Oops, fix breakage from previous commit