Fixed a bug with the parallelization