From 4a508223cdfdc0f60e9e05a373ec301e7d4444a6 Mon Sep 17 00:00:00 2001
From: Tinashe Musonza <tinashemusonza@gmail.com>
Date: Mon, 23 Oct 2023 21:25:54 -0400
Subject: [PATCH] wip

---
 src/DynamoBreezeService.php          | 1 +
 tests/Feature/ExampleFeatureTest.php | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/DynamoBreezeService.php b/src/DynamoBreezeService.php
index 6ae0c88..22c5d95 100644
--- a/src/DynamoBreezeService.php
+++ b/src/DynamoBreezeService.php
@@ -130,6 +130,7 @@ public function batchGet(array $batchItems): DynamoBreezeResult
         $requestItems = [];
 
         foreach ($batchItems as $tableIdentifier => $details) {
+            $this->tableIdentifier = $tableIdentifier;
             $keys = [];
             foreach ($details['keys'] as $keyData) {
                 $keys[] = $this->expressionAttributeHandler->marshaler->marshalItem($keyData);
diff --git a/tests/Feature/ExampleFeatureTest.php b/tests/Feature/ExampleFeatureTest.php
index 843f034..a8b1220 100644
--- a/tests/Feature/ExampleFeatureTest.php
+++ b/tests/Feature/ExampleFeatureTest.php
@@ -145,8 +145,7 @@ public function testBatchGetUserPosts(): void
             ['PostId' => '1', 'Timestamp' => 11111],
         ];
 
-        $result = DynamoBreeze::withTableIdentifier(self::TABLE_IDENTIFIER)
-            ->batchGet([
+        $result = DynamoBreeze::batchGet([
                 self::TABLE_IDENTIFIER => [
                     'keys' => $keysToGet,
                 ],