Update bucket arg check

Tiny error of omission...