diff --git a/influxdb/tests/dataframe_client_test.py b/influxdb/tests/dataframe_client_test.py
index f6db3c22..313891b0 100644
--- a/influxdb/tests/dataframe_client_test.py
+++ b/influxdb/tests/dataframe_client_test.py
@@ -20,7 +20,10 @@
 
 if not using_pypy:
     import pandas as pd
-    from pandas.util.testing import assert_frame_equal
+    try:
+        from pandas.testing import assert_frame_equal
+    except ImportError:
+        from pandas.util.testing import assert_frame_equal
     from influxdb import DataFrameClient
     import numpy as np