diff --git a/utils.py b/utils.py
index 2acd663..ab18a6f 100644
--- a/utils.py
+++ b/utils.py
@@ -4,6 +4,7 @@ import torch.optim as optim
 from PIL import Image
 from torchvision import models
 import numpy as np
+import PIL
 
 def get_model(config):
     if config.dataset == 'gender_dataset':