Add test for bucket selection