diff --git a/client/src/app/routes/GameRoutes.tsx b/client/src/app/routes/GameRoutes.tsx index c8a1415fe..9783af583 100644 --- a/client/src/app/routes/GameRoutes.tsx +++ b/client/src/app/routes/GameRoutes.tsx @@ -278,38 +278,42 @@ function SongInfoHeader({ ) ); - let ImageCell = null; + const [ImageCell, setImageCell] = useState(null); - if (game === "popn") { - ImageCell = ( - ).data.inGameID - }.png` - )} - className="w-100" - /> - ); - } else if (game === "itg") { - const banner = (song as SongDocument<"itg">).data.banner; - - if (banner) { - ImageCell = ( + useEffect(() => { + if (game === "popn") { + setImageCell( ).data.inGameID + }.png` + )} + className="w-100" + /> + ); + } else if (game === "itg") { + const banner = (song as SongDocument<"itg">).data.banner; + + if (banner) { + setImageCell( + + ); + } + } else if ("displayVersion" in song.data) { + setImageCell( + setImageCell(null)} + src={ToCDNURL(`/game-icons/${game}/${song.data.displayVersion}`)} className="w-100" /> ); } - } else if ("displayVersion" in song.data) { - ImageCell = ( - - ); - } + }, [game, song]); return (