diff --git a/src/snowflake/snowpark/functions.py b/src/snowflake/snowpark/functions.py
index 4fc28e683c..2762b4169b 100644
--- a/src/snowflake/snowpark/functions.py
+++ b/src/snowflake/snowpark/functions.py
@@ -5377,11 +5377,13 @@ def array_remove(array: ColumnOrName, element: ColumnOrLiteral) -> Column:
         -------------
         <BLANKLINE>
 
-        >>> df.select(array_remove(array_remove(array_remove(df.data, 1), 3.1), lit('2').cast(VariantType())).alias("objects")).show()
+        >>> df.select(array_remove(array_remove(df.data, 1), lit('2').cast(VariantType())).alias("objects")).show()
         -------------
         |"OBJECTS"  |
         -------------
-        |[]         |
+        |[          |
+        |  3.1      |
+        |]          |
         -------------
         <BLANKLINE>