Fix weird multiprocessing bug