From 28b9886b061d0fec2d10a28377c5c55bf6af95dd Mon Sep 17 00:00:00 2001
From: Mitchell McCaffrey <mitchemmc@gmail.com>
Date: Wed, 20 Jan 2021 18:08:03 +1100
Subject: [PATCH] Moved asset load callbacks to take into account database
 insertion time

---
 src/network/NetworkedMapAndTokens.js | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/network/NetworkedMapAndTokens.js b/src/network/NetworkedMapAndTokens.js
index d5e5075..ee03596 100644
--- a/src/network/NetworkedMapAndTokens.js
+++ b/src/network/NetworkedMapAndTokens.js
@@ -370,8 +370,9 @@ function NetworkedMapAndTokens({ session }) {
 
       if (id === "mapResponse") {
         const newMap = data;
-        await putMap(newMap);
         setCurrentMap(newMap);
+        await putMap(newMap);
+        assetLoadFinish();
       }
 
       if (id === "tokenRequest") {
@@ -382,6 +383,7 @@ function NetworkedMapAndTokens({ session }) {
       if (id === "tokenResponse") {
         const newToken = data;
         await putToken(newToken);
+        assetLoadFinish();
       }
     }
 
@@ -389,9 +391,6 @@ function NetworkedMapAndTokens({ session }) {
       if (count === 1) {
         assetLoadStart();
       }
-      if (total === count) {
-        assetLoadFinish();
-      }
       assetProgressUpdate({ id, total, count });
     }