Finalize #51 fix