diff --git a/CHANGELOG.md b/CHANGELOG.md
index 5a6eed1d..c8d671fe 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -31,6 +31,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Added missing `weights_only=False` argument to fix loading train checkpoints with newer versions of PyTorch.
 - Fixed bug where GCS upload does not retry on transient failures.
+- Fixed bug in `get_mock_batch()` by adding `non_cuda` option so that mock batches can be loaded on cuda devices in addition to MPS devices.
 
 ## [v1.7.0](https://github.com/allenai/OLMo-core/releases/tag/v1.7.0) - 2024-11-27