diff --git a/content/courses/intro-to-solana/interact-with-wallets.md b/content/courses/intro-to-solana/interact-with-wallets.md
index 53cdae15f..76cfc9db1 100644
--- a/content/courses/intro-to-solana/interact-with-wallets.md
+++ b/content/courses/intro-to-solana/interact-with-wallets.md
@@ -113,7 +113,7 @@ ensure this is to wrap your entire app in `ConnectionProvider` and
 
 ```tsx
 import { NextPage } from "next";
-import { FC, ReactNode } from "react";
+import { FC, ReactNode, useMemo } from "react";
 import {
   ConnectionProvider,
   WalletProvider,
@@ -159,7 +159,7 @@ full-featured wallet experience is to use `WalletModalProvider` and
 
 ```tsx
 import { NextPage } from "next";
-import { FC, ReactNode } from "react";
+import { FC, ReactNode, useMemo } from "react";
 import {
   ConnectionProvider,
   WalletProvider,
@@ -267,7 +267,7 @@ export const BalanceDisplay: FC = () => {
 
   return (
     <div>
-      <p>{publicKey ? `Balance: ${balance / LAMPORTS_PER_SOL} SOL` : ""}</p>
+      <p>{publicKey ? `Balance: ${balance} SOL` : ""}</p>
     </div>
   );
 };
@@ -426,7 +426,7 @@ To complete this component, add
 ensure proper styling and behavior of the Wallet Adapter library components.
 
 ```tsx
-import { FC, ReactNode } from "react";
+import { FC, ReactNode, useMemo } from "react";
 import {
   ConnectionProvider,
   WalletProvider,