diff --git a/drivers/openstack/openstack.go b/drivers/openstack/openstack.go
index f0e01b0..d91ac2b 100644
--- a/drivers/openstack/openstack.go
+++ b/drivers/openstack/openstack.go
@@ -218,6 +218,10 @@ func (d *Driver) GetSSHHostname() (string, error) {
 	return d.GetIP()
 }
 
+func (d *Driver) SetClient(client Client) {
+	d.client = client
+}
+
 func (d *Driver) DriverName() string {
 	return "openstack"
 }