Fix multi-thread issue