Fix test reload bug