diff --git a/python/fbprophet/tests/test_diagnostics.py b/python/fbprophet/tests/test_diagnostics.py
index 1331873..ae59e67 100644
--- a/python/fbprophet/tests/test_diagnostics.py
+++ b/python/fbprophet/tests/test_diagnostics.py
@@ -112,7 +112,8 @@ class TestDiagnostics(TestCase):
             m.fit(df)
             df_cv = diagnostics.cross_validation(
                 m, horizon='4 days', period='4 days', initial='115 days')
-            self.assertListEqual(['ds', 'yhat', 'y', 'cutoff'], df_cv.columns.tolist())
+            expected_cols = ['ds', 'yhat', 'y', 'cutoff']
+            self.assertTrue(all(col in expected_cols for col in df_cv.columns.tolist()))
 
     def test_performance_metrics(self):
         m = Prophet()
diff --git a/python/fbprophet/tests/test_prophet.py b/python/fbprophet/tests/test_prophet.py
index 31ef790..acb0e33 100644
--- a/python/fbprophet/tests/test_prophet.py
+++ b/python/fbprophet/tests/test_prophet.py
@@ -102,9 +102,8 @@ class TestProphet(TestCase):
             m = Prophet(uncertainty_samples=uncertainty)
             m.fit(train)
             fcst = m.predict(future)
-            self.assertListEqual(['ds', 'trend', 'additive_terms', 'weekly',
-                                  'multiplicative_terms', 'yhat'],
-                                 fcst.columns.tolist())
+            expected_cols = ['ds', 'trend', 'additive_terms', 'multiplicative_terms', 'weekly', 'yhat']
+            self.assertTrue(all(col in expected_cols for col in fcst.columns.tolist()))
 
     def test_setup_dataframe(self):
         m = Prophet()