diff --git a/tests/test_diff_tally/test_diff_tally.py b/tests/test_diff_tally/test_diff_tally.py
index ecf281034..c5a0762a6 100644
--- a/tests/test_diff_tally/test_diff_tally.py
+++ b/tests/test_diff_tally/test_diff_tally.py
@@ -2,7 +2,10 @@
 
 import glob
 import os
-from io import StringIO
+try:
+    from StringIO import StringIO
+except:
+    from io import StringIO
 import sys
 
 import pandas as pd