fix assertion issue due to multiprocess