diff --git a/dbclient/WorkspaceClient.py b/dbclient/WorkspaceClient.py
index c5f115d..d06d78a 100644
--- a/dbclient/WorkspaceClient.py
+++ b/dbclient/WorkspaceClient.py
@@ -351,7 +351,7 @@ def download_notebook_helper(self, notebook_data, checkpoint_notebook_set, error
             logging_utils.log_response_error(error_logger, resp)
             return resp
         if resp.get('error_code', None):
-            if self.skip_large_nb and resp.message == 'Size exceeds 10485760 bytes':
+            if self.skip_large_nb and resp.get(resp.message, None) == 'Size exceeds 10485760 bytes':
                 logging.info("Notebook {} skipped due to size exceeding limit".format(notebook_path))
             else:
                 resp['path'] = notebook_path