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): '''