diff --git a/dbclient/WorkspaceClient.py b/dbclient/WorkspaceClient.py
index ffd61cc..331b83b 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.get("message", None) == 'Size exceeds 10485760 bytes':
+            if self.skip_large_nb and re.search(r"Notebook size exceeded limit: \d+ > 10485760", resp.get("message", '')):
                 logging.info("Notebook {} skipped due to size exceeding limit".format(notebook_path))
             else:
                 resp['path'] = notebook_path