diff --git a/background_task/models.py b/background_task/models.py
index f690691..513da96 100644
--- a/background_task/models.py
+++ b/background_task/models.py
@@ -5,8 +5,12 @@
 from datetime import timedelta
 from hashlib import sha1
 import traceback
-from StringIO import StringIO
 import logging
+import sys
+if sys.version_info[0] < 3:
+    from StringIO import StringIO
+else:
+    from io import StringIO
 
 try:
     import json  # Django >= 1.6