diff --git a/spring-graphql/src/main/resources/META-INF/native-image/org.springframework.graphql/spring-graphql/reflect-config.json b/spring-graphql/src/main/resources/META-INF/native-image/org.springframework.graphql/spring-graphql/reflect-config.json
index fab40865..98b48667 100644
--- a/spring-graphql/src/main/resources/META-INF/native-image/org.springframework.graphql/spring-graphql/reflect-config.json
+++ b/spring-graphql/src/main/resources/META-INF/native-image/org.springframework.graphql/spring-graphql/reflect-config.json
@@ -5,7 +5,16 @@
     "allDeclaredMethods":true,
     "allDeclaredConstructors":true,
     "condition": {
-      "typeReachable": "org.springframework.graphql.client.WebSocketGraphQlTransport"
+      "typeReachable": "org.springframework.graphql.client.DefaultRSocketGraphQlClientBuilder"
+    }
+  },
+  {
+    "name":"org.springframework.graphql.server.support.GraphQlWebSocketMessage",
+    "allDeclaredFields":true,
+    "allDeclaredMethods":true,
+    "allDeclaredConstructors":true,
+    "condition": {
+      "typeReachable": "org.springframework.graphql.client.DefaultWebSocketGraphQlClientBuilder"
     }
   }
 ]