ENH: fix reload bug