diff --git a/test/livebook/session/data_test.exs b/test/livebook/session/data_test.exs
index 55a2a8a86..58e87b565 100644
--- a/test/livebook/session/data_test.exs
+++ b/test/livebook/session/data_test.exs
@@ -2295,7 +2295,8 @@ defmodule Livebook.Session.DataTest do
 
       empty_map = %{}
 
-      assert {:ok, %{input_values: ^empty_map}, _} = Data.apply_operation(data, operation)
+      assert {:ok, %{input_values: ^empty_map}, [{:clean_up_input_values, %{"i1" => "value"}}]} =
+               Data.apply_operation(data, operation)
     end
 
     test "does not garbage collect inputs if present in another cell" do