diff --git a/tests/basic_deployment.py b/tests/basic_deployment.py
index 9cf5516e..045d7225 100755
--- a/tests/basic_deployment.py
+++ b/tests/basic_deployment.py
@@ -3,6 +3,7 @@
 import amulet
 import types
 from time import sleep
+import yaml
 import cinderclient.v1.client as cinder_client
 
 from charmhelpers.contrib.openstack.amulet.deployment import (