Fix timing issue on test.