Use Capabilities rather than ImmutableCapabilities in mutators

TikhomirovSergey · TikhomirovSergey · commit fbb6e58eb5d2 · 2018-02-09T23:12:24.000+03:00
diff --git a/src/main/java/org/openqa/selenium/remote/NewSessionPayload.java b/src/main/java/org/openqa/selenium/remote/NewSessionPayload.java
@@ -221,7 +221,8 @@ public void writeTo(Appendable appendable) throws IOException {
 
             Map<String, Object> first = getOss();
             if (first == null) {
-                first = stream().findFirst()
+                //noinspection unchecked
+                first = (Map<String, Object>) stream().findFirst()
                         .orElse(new ImmutableCapabilities())
                         .asMap();
             }
@@ -309,7 +310,7 @@ private void streamW3CProtocolParameters(JsonOutput out, Map<String, Object> des
      * equivalent W3C capabilities isn't particularly easy, so it's hoped that this approach gives us
      * the most compatible implementation.
      */
-    public Stream<ImmutableCapabilities> stream() throws IOException {
+    public Stream<Capabilities> stream() throws IOException {
         // OSS first
         Stream<Map<String, Object>> oss = Stream.of(getOss());
 
