[TSan] Fixed a race