diff --git a/src/Interpreters/MutationsInterpreter.cpp b/src/Interpreters/MutationsInterpreter.cpp
index df0267b9450..985fda3aac7 100644
--- a/src/Interpreters/MutationsInterpreter.cpp
+++ b/src/Interpreters/MutationsInterpreter.cpp
@@ -36,34 +36,44 @@ namespace ErrorCodes
 
 namespace
 {
-struct FirstNonDeterministicFuncData
+/// Helps to detect situations, where non-deterministic functions may be used in mutations of Replicated*MergeTree.
+class FirstNonDeterministicFuncMatcher
 {
-    using TypeToVisit = ASTFunction;
+public:
+    struct Data {
+        const Context & context;
+        std::optional<String> nondeterministic_function_name;
+    };
 
-    explicit FirstNonDeterministicFuncData(const Context & context_)
-        : context{context_}
-    {}
-
-    const Context & context;
-    std::optional<String> nondeterministic_function_name;
-
-    void visit(ASTFunction & function, ASTPtr &)
+public:
+    static bool needChildVisit(const ASTPtr & /*node*/, const ASTPtr & child)
     {
-        if (nondeterministic_function_name)
+        return child != nullptr;
+    }
+
+    static void visit(const ASTPtr & node, Data & data)
+    {
+        if (data.nondeterministic_function_name)
             return;
 
-        const auto func = FunctionFactory::instance().get(function.name, context);
-        if (!func->isDeterministic())
-            nondeterministic_function_name = func->getName();
+        if (const auto * function = typeid_cast<const ASTFunction *>(node.get()))
+        {
+            if (function->name != "lambda")
+            {
+                const auto func = FunctionFactory::instance().get(function->name, data.context);
+                if (!func->isDeterministic())
+                    data.nondeterministic_function_name = func->getName();
+            }
+        }
     }
 };
 
 using FirstNonDeterministicFuncFinder =
-        InDepthNodeVisitor<OneTypeMatcher<FirstNonDeterministicFuncData>, true>;
+        InDepthNodeVisitor<FirstNonDeterministicFuncMatcher, true>;
 
 std::optional<String> findFirstNonDeterministicFuncName(const MutationCommand & command, const Context & context)
 {
-    FirstNonDeterministicFuncData finder_data(context);
+    FirstNonDeterministicFuncMatcher::Data finder_data{context};
 
     switch (command.type)
     {
diff --git a/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.reference b/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.reference
index f799e8ed8f0..6bf25043399 100644
--- a/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.reference
+++ b/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.reference
@@ -5,3 +5,5 @@ OK
 OK
 OK
 OK
+OK
+OK
diff --git a/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.sh b/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.sh
index 9b190855adf..68cb5e0e760 100755
--- a/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.sh
+++ b/tests/queries/0_stateless/01017_mutations_with_nondeterministic_functions_zookeeper.sh
@@ -43,6 +43,12 @@ ${CLICKHOUSE_CLIENT} --query "ALTER TABLE $R1 DELETE WHERE ignore(rand())" 2>&1
 ${CLICKHOUSE_CLIENT} --query "ALTER TABLE $R1 UPDATE y = y + rand() % 1 WHERE not ignore()" 2>&1 \
 | fgrep -q "must use only deterministic functions" && echo 'OK' || echo 'FAIL'
 
+${CLICKHOUSE_CLIENT} --query "ALTER TABLE $R1 UPDATE y = x + arrayCount(x -> (x + y) % 2, range(y)) WHERE not ignore()" 2>&1 > /dev/null \
+&& echo 'OK' || echo 'FAIL'
+
+${CLICKHOUSE_CLIENT} --query "ALTER TABLE $R1 UPDATE y = x + arrayCount(x -> (rand() + x) % 2, range(y)) WHERE not ignore()" 2>&1 \
+| fgrep -q "must use only deterministic functions" && echo 'OK' || echo 'FAIL'
+
 
 # For regular tables we do not enforce deterministic functions
 ${CLICKHOUSE_CLIENT} --query "ALTER TABLE $T1 DELETE WHERE rand() = 0" 2>&1 > /dev/null \