diff --git a/tests/migrations/test_writer.py b/tests/migrations/test_writer.py
index 8febf40f03..ad77b51ae7 100644
--- a/tests/migrations/test_writer.py
+++ b/tests/migrations/test_writer.py
@@ -7,6 +7,7 @@ import functools
 import math
 import os
 import re
+import sys
 import tokenize
 import unittest
 
@@ -35,6 +36,8 @@ try:
 except ImportError:
     enum = None
 
+PY36 = sys.version_info >= (3, 6)
+
 
 class Money(decimal.Decimal):
     def deconstruct(self):
@@ -412,7 +415,10 @@ class WriterTests(SimpleTestCase):
         # Test a string regex with flag
         validator = RegexValidator(r'^[0-9]+$', flags=re.U)
         string = MigrationWriter.serialize(validator)[0]
-        self.assertEqual(string, "django.core.validators.RegexValidator('^[0-9]+$', flags=32)")
+        if PY36:
+            self.assertEqual(string, "django.core.validators.RegexValidator('^[0-9]+$', flags=re.RegexFlag(32))")
+        else:
+            self.assertEqual(string, "django.core.validators.RegexValidator('^[0-9]+$', flags=32)")
         self.serialize_round_trip(validator)
 
         # Test message and code