diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py
index e73fe65..d240f1e 100644
--- a/deepface/models/Demography.py
+++ b/deepface/models/Demography.py
@@ -43,6 +43,6 @@ class Demography(ABC):
         # Check input dimension
         if len(image_batch.shape) == 3:
             # Single image - add batch dimension
-            imgs = np.expand_dims(image_batch, axis=0)
+            image_batch = np.expand_dims(image_batch, axis=0)
 
         return image_batch