diff --git a/test/test_cirros.py b/test/test_cirros.py
index 24b5c4e..1451d11 100644
--- a/test/test_cirros.py
+++ b/test/test_cirros.py
@@ -20,7 +20,7 @@
 import requests
 import subprocess
 import time
-
+import yaml
 import certs
 
 logger = logging.getLogger(__name__)