Fix broken unit test from parallism case