Add unit tests for for race