diff --git a/nfnets/agc.py b/nfnets/agc.py
index 5867220..abe1f89 100644
--- a/nfnets/agc.py
+++ b/nfnets/agc.py
@@ -2,7 +2,7 @@
 from torch import nn, optim
 
 from nfnets.utils import unitwise_norm
-from collections import Iterable
+from collections.abc import Iterable
 
 
 class AGC(optim.Optimizer):