From 453fc4c2f53d065f2c4459ac9eeb16031ccf7430 Mon Sep 17 00:00:00 2001
From: Durran Jordan <durran@gmail.com>
Date: Thu, 29 Jul 2021 17:03:54 +0200
Subject: [PATCH] test: fix session pinning assertions

---
 test/functional/unified-spec-runner/operations.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/functional/unified-spec-runner/operations.ts b/test/functional/unified-spec-runner/operations.ts
index 43bb4ac2be..cad0e0cd62 100644
--- a/test/functional/unified-spec-runner/operations.ts
+++ b/test/functional/unified-spec-runner/operations.ts
@@ -127,12 +127,12 @@ operations.set('assertSessionNotDirty', async ({ entities, operation }) => {
 });
 
 operations.set('assertSessionPinned', async ({ entities, operation }) => {
-  const session = entities.getEntity('session', operation.arguments.session);
+  const session = operation.arguments.session;
   expect(session.isPinned, 'session should be pinned').to.be.true;
 });
 
 operations.set('assertSessionUnpinned', async ({ entities, operation }) => {
-  const session = entities.getEntity('session', operation.arguments.session);
+  const session = operation.arguments.session;
   expect(session.isPinned, 'session should be unpinned').to.be.false;
 });