diff --git a/network/Resnet50.py b/network/Resnet50.py
index 5d52200df2f8b0e45b790aa85c4f1858d49993ff..206aaf255ef6314a01c376a7032cc748f2d58128 100644
--- a/network/Resnet50.py
+++ b/network/Resnet50.py
@@ -1,5 +1,6 @@
 from tensorflow.keras import regularizers
 
+from helpers.CAMModel import CAMModel
 from network.NetworkBase import NetworkBase
 from tensorflow.keras.models import Model
 from tensorflow.keras.layers import Input, Dense, GlobalAveragePooling2D, Dropout, Flatten, AveragePooling2D