Fixing missing import in buckets

Don't know why I got these errors today.