Fix test for assign_pid