Use six to reload module

1 files changed, 3 insertions, 2 deletions

diff --git a/rest_framework/tests/utils.py b/rest_framework/tests/utils.py
index ee157824..a8f2eb0b 100644
--- a/rest_framework/tests/utils.py
+++ b/rest_framework/tests/utils.py
@@ -1,4 +1,5 @@
 from contextlib import contextmanager
+from rest_framework.compat import six
 from rest_framework.settings import api_settings
 
 
@@ -14,11 +15,11 @@ def temporary_setting(setting, value, module=None):
     setattr(api_settings, setting, value)
 
     if module is not None:
-        reload(module)
+        six.moves.reload_module(module)
 
     yield
 
     setattr(api_settings, setting, original_value)
 
     if module is not None:
-        reload(module)
+        six.moves.reload_module(module)