fix: buckets test
tests: fixes in tests after refactoring