diff --git a/vector_quantize_pytorch/residual_fsq.py b/vector_quantize_pytorch/residual_fsq.py
index 82753a3..6c19310 100644
--- a/vector_quantize_pytorch/residual_fsq.py
+++ b/vector_quantize_pytorch/residual_fsq.py
@@ -1,5 +1,5 @@
 import random
-from math import log2
+from math import ceil
 from functools import partial
 
 from typing import List
diff --git a/vector_quantize_pytorch/residual_lfq.py b/vector_quantize_pytorch/residual_lfq.py
index 475025b..bf4c3f0 100644
--- a/vector_quantize_pytorch/residual_lfq.py
+++ b/vector_quantize_pytorch/residual_lfq.py
@@ -1,5 +1,5 @@
 import random
-from math import log2
+from math import ceil, log2
 from functools import partial, cache
 
 import torch