diff --git a/test/archethic_web/graphql_schema_test.exs b/test/archethic_web/graphql_schema_test.exs
index 4bcc3b94ee..fedbba6cf8 100644
--- a/test/archethic_web/graphql_schema_test.exs
+++ b/test/archethic_web/graphql_schema_test.exs
@@ -347,6 +347,9 @@ defmodule ArchethicWeb.GraphQLSchemaTest do
 
         _, %GetGenesisAddress{}, _ ->
           {:ok, %NotFound{}}
+
+        _, %GetLastTransactionAddress{address: address}, _ ->
+          {:ok, %LastTransactionAddress{address: address}}
       end)
 
       conn =
@@ -473,8 +476,12 @@ defmodule ArchethicWeb.GraphQLSchemaTest do
       addr = <<0::8, 0::8, :crypto.strong_rand_bytes(32)::binary>>
 
       MockClient
-      |> stub(:send_message, fn _, %GetBalance{}, _ ->
-        {:ok, %Balance{uco: 218_000_000}}
+      |> stub(:send_message, fn
+        _, %GetBalance{}, _ ->
+          {:ok, %Balance{uco: 218_000_000}}
+
+        _, %GetLastTransactionAddress{address: address}, _ ->
+          {:ok, %LastTransactionAddress{address: address}}
       end)
 
       conn =
@@ -489,15 +496,19 @@ defmodule ArchethicWeb.GraphQLSchemaTest do
       addr = <<0::8, 0::8, :crypto.strong_rand_bytes(32)::binary>>
 
       MockClient
-      |> stub(:send_message, fn _, %GetBalance{}, _ ->
-        {:ok,
-         %Balance{
-           token: %{
-             {"@Token1", 0} => 200_000_000,
-             {"@Token2", 0} => 500_000_000,
-             {"@Token3", 0} => 1_000_000_000
-           }
-         }}
+      |> stub(:send_message, fn
+        _, %GetBalance{}, _ ->
+          {:ok,
+           %Balance{
+             token: %{
+               {"@Token1", 0} => 200_000_000,
+               {"@Token2", 0} => 500_000_000,
+               {"@Token3", 0} => 1_000_000_000
+             }
+           }}
+
+        _, %GetLastTransactionAddress{address: address}, _ ->
+          {:ok, %LastTransactionAddress{address: address}}
       end)
 
       conn =
@@ -549,6 +560,11 @@ defmodule ArchethicWeb.GraphQLSchemaTest do
     test "should return same address", %{conn: conn} do
       addr = <<0::8, 0::8, :crypto.strong_rand_bytes(32)::binary>>
 
+      MockClient
+      |> stub(:send_message, fn _, %GetGenesisAddress{}, _ ->
+        {:ok, %GenesisAddress{address: addr}}
+      end)
+
       conn =
         post(conn, "/api", %{
           "query" => "query {