diff --git a/modules/main/src/main/java/com/annimon/ownlang/modules/functional/functional_stream.java b/modules/main/src/main/java/com/annimon/ownlang/modules/functional/functional_stream.java
index 5ba6d70..2017326 100644
--- a/modules/main/src/main/java/com/annimon/ownlang/modules/functional/functional_stream.java
+++ b/modules/main/src/main/java/com/annimon/ownlang/modules/functional/functional_stream.java
@@ -9,18 +9,11 @@ public final class functional_stream implements Function {
     public Value execute(Value[] args) {
         Arguments.checkAtLeast(1, args.length);
 
-        if (args.length > 1) {
-            return new StreamValue(new ArrayValue(args));
-        }
-
         final Value value = args[0];
-        switch (value.type()) {
-            case Types.MAP:
-                return new StreamValue(((MapValue) value).toPairs());
-            case Types.ARRAY:
-                return new StreamValue((ArrayValue) value);
-            default:
-                throw new TypeException("Invalid argument. Array or map expected");
-        }
+        return switch (value.type()) {
+            case Types.MAP -> new StreamValue(((MapValue) value).toPairs());
+            case Types.ARRAY -> new StreamValue((ArrayValue) value);
+            default -> throw new TypeException("Invalid argument. Array or map expected");
+        };
     }
 }