diff --git a/extensions-builtin/forge_space_florence_2/forge_app.py b/extensions-builtin/forge_space_florence_2/forge_app.py
index bdb99645..de6034f4 100644
--- a/extensions-builtin/forge_space_florence_2/forge_app.py
+++ b/extensions-builtin/forge_space_florence_2/forge_app.py
@@ -11,6 +11,7 @@ import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 
 import random
+import numpy as np
 
 from modules import shared