diff --git a/tests/policies/base__test_batch.pkl b/tests/policies/base__test_batch.pkl
index 8bc8715..756daf2 100644
Binary files a/tests/policies/base__test_batch.pkl and b/tests/policies/base__test_batch.pkl differ
diff --git a/tests/policies/base__transforms.pkl b/tests/policies/base__transforms.pkl
index 39fbbf7..fa1e4c9 100644
Binary files a/tests/policies/base__transforms.pkl and b/tests/policies/base__transforms.pkl differ