diff --git a/scripts/algorithms/arima.py b/scripts/algorithms/arima.py
index 1c621c6..a660514 100644
--- a/scripts/algorithms/arima.py
+++ b/scripts/algorithms/arima.py
@@ -1,5 +1,5 @@
 from numpy import clip, inf
-from pyramid.arima import auto_arima
+from pmdarima.arima import auto_arima
 
 
 class ARIMAForecast(object):
diff --git a/setup.py b/setup.py
index de94875..e9d6e81 100644
--- a/setup.py
+++ b/setup.py
@@ -54,7 +54,7 @@ def setup_package():
             'Programming Language :: Python :: 3.6',
         ],
         install_requires=['matplotlib', 'numpy', 'scipy', 'wordcloud', 'pandas', 'tqdm', 'nltk', 'scikit-learn', 'xlrd',
-                          'python-Levenshtein', 'gensim', 'pyramid-arima>=0.9.0', 'keras', 'tensorflow', 'keras_tqdm',
+                          'python-Levenshtein', 'gensim', 'pmdarima', 'keras', 'tensorflow', 'keras_tqdm',
                           'patsy', 'humanfriendly', 'psutil', 'jinja2'],
         # extras_require={'dev': ['check-manifest'],'test': ['coverage'],},
         python_requires='>=3.6',
diff --git a/tests/algorithms/test_arima.py b/tests/algorithms/test_arima.py
index 04769bf..8ba0b79 100644
--- a/tests/algorithms/test_arima.py
+++ b/tests/algorithms/test_arima.py
@@ -39,4 +39,4 @@ def test_linearly_increasing_sequence(self):
 
         actual_prediction = arima.predict_counts()
 
-        np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=1)
+        np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=0)