diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py
index ddb65f9..49b5270 100644
--- a/metastable_baselines/misc/rollout_buffer.py
+++ b/metastable_baselines/misc/rollout_buffer.py
@@ -1,5 +1,4 @@
 from typing import Any, Dict, Optional, Type, Union, NamedTuple
-from more_itertools import distribute
 
 import numpy as np
 import torch as th