diff --git a/gopls/internal/regtest/misc/semantictokens_test.go b/gopls/internal/regtest/misc/semantictokens_test.go
index 8921995cf7..cd9dce35c6 100644
--- a/gopls/internal/regtest/misc/semantictokens_test.go
+++ b/gopls/internal/regtest/misc/semantictokens_test.go
@@ -93,8 +93,7 @@ func Add[T int](target T, l []T) []T {
 		Settings{"semanticTokens": true},
 	).Run(t, src, func(t *testing.T, env *Env) {
 		env.OpenFile("main.go")
-		//env.Await(EmptyOrNoDiagnostics("main.go`"))
-		env.Await(env.AfterChange(env.DiagnosticAtRegexp("main.go", "for range")))
+		env.AfterChange(env.DiagnosticAtRegexp("main.go", "for range"))
 		p := &protocol.SemanticTokensParams{
 			TextDocument: protocol.TextDocumentIdentifier{
 				URI: env.Sandbox.Workdir.URI("main.go"),