diff --git a/virtual/environment_test.go b/virtual/environment_test.go
index eb1bc0c..6ce95b3 100644
--- a/virtual/environment_test.go
+++ b/virtual/environment_test.go
@@ -1095,6 +1095,8 @@ func (ta *testActor) Invoke(
 	operation string,
 	payload []byte,
 ) ([]byte, error) {
+	// Check the context immediately to ensure we catch bugs related to
+	// canceling contexts prematurely.
 	if err := ctx.Err(); err != nil {
 		return nil, ctx.Err()
 	}