diff --git a/regression/storage.py b/regression/storage.py index 7ec1cac749ba..bc7c3b1980a4 100644 --- a/regression/storage.py +++ b/regression/storage.py @@ -55,7 +55,7 @@ def setUp(self): def tearDown(self): with Batch(CONNECTION) as batch: for bucket_name in self.case_buckets_to_delete: - batch.get_bucket(bucket_name).delete() + storage.Bucket(connection=batch, name=bucket_name).delete() def test_create_bucket(self): new_bucket_name = 'a-new-bucket'