diff --git a/flash/core/data/utils.py b/flash/core/data/utils.py
index 380253789c..2d11f1bea3 100644
--- a/flash/core/data/utils.py
+++ b/flash/core/data/utils.py
@@ -37,7 +37,7 @@ def download_file(url: str, path: str, verbose: bool = False) -> None:
 
     if not os.path.exists(local_filename):
         r = requests.get(url, stream=True)
-        file_size = int(r.headers['Content-Length']) if 'Content-Length' in r.headers else 0
+        file_size = int(r.headers.get('Content-Length', 0))
         chunk = 1
         chunk_size = 1024
         num_bars = int(file_size / chunk_size)