Add some setup to check if it corrects the race condition