diff --git a/medium-gcp-ai-platform/ai-platform-train/setup.py b/medium-gcp-ai-platform/ai-platform-train/setup.py
index 6fefc38..0a22d49 100644
--- a/medium-gcp-ai-platform/ai-platform-train/setup.py
+++ b/medium-gcp-ai-platform/ai-platform-train/setup.py
@@ -7,7 +7,7 @@
                 'lightgbm==3.1.1',
                 'cloudpickle==1.6.0',
                 'gcsfs==0.7.1',
-                'astropy==4.1']
+                'astropy==5.3.3']
 
 setup(name='har_model',
       install_requires=DEPENDENCIES,