diff --git a/molecule/ec2-centos-7/tests/test_default.py b/molecule/ec2-centos-7/tests/test_default.py
index cdf887f..3ddb355 100644
--- a/molecule/ec2-centos-7/tests/test_default.py
+++ b/molecule/ec2-centos-7/tests/test_default.py
@@ -1,4 +1,5 @@
 import os
+import yaml
 
 import testinfra.utils.ansible_runner