diff --git a/tests/serializers/test_yaml.py b/tests/serializers/test_yaml.py
index dbcbd57003..10f73901cb 100644
--- a/tests/serializers/test_yaml.py
+++ b/tests/serializers/test_yaml.py
@@ -116,7 +116,7 @@ class YamlSerializerTestCase(SerializersTestBase, TestCase):
     headline: Poker has no place on ESPN
     pub_date: 2006-06-16 11:00:00
     categories:""" + (
-        ' [%(first_category_pk)s, %(second_category_pk)s]' if yaml.__version__ < '5.1'
+        ' [%(first_category_pk)s, %(second_category_pk)s]' if HAS_YAML and yaml.__version__ < '5.1'
         else '\n    - %(first_category_pk)s\n    - %(second_category_pk)s') + """
     meta_data: []
 """