diff --git a/YOLOv8_Explainer/core.py b/YOLOv8_Explainer/core.py
index 6dd099e..b0c8675 100644
--- a/YOLOv8_Explainer/core.py
+++ b/YOLOv8_Explainer/core.py
@@ -177,7 +177,7 @@ def __init__(
         Initialize the YOLOv8 heatmap layer.
         """
         device = device
-        self.backward_type = "all"
+        backward_type = "all"
         ckpt = torch.load(weight)
         model_names = ckpt['model'].names
         model = attempt_load_weights(weight, device)