Python error: ImportError: No module named decorator

I ran into a similar issue when testing out the latest Hail version on the google cloud. I just tried to run the count script from Using Hail on the Google Cloud Platform with a small vds:

Traceback (most recent call last):
File “/tmp/4fb6177b-679b-459a-a5b0-d370f245de96/count_test.py”, line 1, in
from hail import *
File “/home/ec2-user/BuildAgent/work/c38e75e72b769a7c/python/hail/init.py”, line 1, in
File “/home/ec2-user/BuildAgent/work/c38e75e72b769a7c/python/hail/context.py”, line 5, in
File “/home/ec2-user/BuildAgent/work/c38e75e72b769a7c/python/hail/dataset.py”, line 3, in
File “/home/ec2-user/BuildAgent/work/c38e75e72b769a7c/python/hail/java.py”, line 2, in
ImportError: No module named decorator

How can I fix this for the google cloud?

Thanks,
Carolina