diff --git a/plugins/modules/organize_bucket.py b/plugins/modules/organize_bucket.py
index 8f4753f..96f8791 100644
--- a/plugins/modules/organize_bucket.py
+++ b/plugins/modules/organize_bucket.py
@@ -2,7 +2,7 @@
 # pylint: disable=missing-module-docstring
 
 from ansible.module_utils.basic import AnsibleModule
-from ansible_collections.tbauriedel.influxdb.plugins.module_utils.utils import (Influx2Api) # pylint: disable=import-error
+from ansible_collections.tbauriedel.influxdb2.plugins.module_utils.utils import (Influx2Api) # pylint: disable=import-error
 
 def run_module():
     '''
diff --git a/roles/influxdb2/tasks/setup.yml b/roles/influxdb2/tasks/setup.yml
index 3fba242..ab50df0 100644
--- a/roles/influxdb2/tasks/setup.yml
+++ b/roles/influxdb2/tasks/setup.yml
@@ -22,7 +22,7 @@
 
 # TODO add handling for orgs, buckets, etc.
 - name: Organize buckets
-  tbauriedel.influxdb.organize_bucket:
+  tbauriedel.influxdb2.organize_bucket:
     name: "{{ item.name }}"
     state: "{{ item.state }}"
     org: "{{ item.org }}"