From 09e6436f80b6c36c8d565f7829c6f569b1a75b8e Mon Sep 17 00:00:00 2001
From: Stefan Sietzen <stefan.sietzen@gmx.at>
Date: Sun, 23 May 2021 12:17:00 +0200
Subject: [PATCH] fixed bug in InstrumentedModel's _unhook_layer

---
 netdissect/nethook.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/netdissect/nethook.py b/netdissect/nethook.py
index 8dffc1d..ae54c53 100644
--- a/netdissect/nethook.py
+++ b/netdissect/nethook.py
@@ -188,7 +188,7 @@ def _unhook_layer(self, aka):
         if aka in self._replacement:
             del self._replacement[aka]
         if aka in self._retained:
-            del self._replacement[aka]
+            del self._retained[aka]
 
     def _postprocess_forward(self, x, aka):
         '''