diff --git a/cloud-ip-add.py b/cloud-ip-add.py
index 508d205..d392cd1 100755
--- a/cloud-ip-add.py
+++ b/cloud-ip-add.py
@@ -3,7 +3,7 @@
 import argparse
 
 from SWSCloudCore import models
-from SWSCloudCore.controllers import ControllerManageIPs
+from SWSCloudCore.controllers.ips.manage import ControllerManageIPs
 
 parser = argparse.ArgumentParser(description='')
 parser.add_argument('--dc', dest="datacenter")
diff --git a/cloud-server-add.py b/cloud-server-add.py
index 0a55fd3..42db758 100755
--- a/cloud-server-add.py
+++ b/cloud-server-add.py
@@ -2,7 +2,7 @@
 
 import argparse
 from uuid import uuid4
-from SWSCloudCore.controllers import ControllerManageServer
+from SWSCloudCore.controllers.servers.manage import ControllerManageServer
 
 parser = argparse.ArgumentParser(description='')
 parser.add_argument('--dc', dest="datacenter")