diff --git a/tests/GmRelay.Bot.Tests/Rendering/SessionBatchViewBuilderTests.cs b/tests/GmRelay.Bot.Tests/Rendering/SessionBatchViewBuilderTests.cs
index 12c9d91..56ea227 100644
--- a/tests/GmRelay.Bot.Tests/Rendering/SessionBatchViewBuilderTests.cs
+++ b/tests/GmRelay.Bot.Tests/Rendering/SessionBatchViewBuilderTests.cs
@@ -66,8 +66,8 @@ public sealed class SessionBatchViewBuilderTests
         var actions = result.Sessions[0].AvailableActions;
 
         Assert.Equal(2, actions.Count);
-        Assert.Equal("join_session", actions[0].ActionKey);
-        Assert.Equal("leave_session", actions[1].ActionKey);
+        Assert.Equal("join_session", result.Sessions[0].AvailableActions[0].ActionKey);
+        Assert.Equal("leave_session", result.Sessions[0].AvailableActions[1].ActionKey);
         Assert.Equal(sessionId, actions[0].SessionId);
         Assert.Equal(sessionId, actions[1].SessionId);
     }
diff --git a/tests/GmRelay.Bot.Tests/Rendering/TelegramSessionBatchRendererTests.cs b/tests/GmRelay.Bot.Tests/Rendering/TelegramSessionBatchRendererTests.cs
index bfcd891..ff05840 100644
--- a/tests/GmRelay.Bot.Tests/Rendering/TelegramSessionBatchRendererTests.cs
+++ b/tests/GmRelay.Bot.Tests/Rendering/TelegramSessionBatchRendererTests.cs
@@ -1,3 +1,4 @@
+using GmRelay.Bot.Infrastructure.Telegram;
 using GmRelay.Shared.Domain;
 using GmRelay.Shared.Rendering;
 using Telegram.Bot.Types.ReplyMarkups;