diff --git a/aws_integration/s3/setup.py b/aws_integration/s3/setup.py
index 1774e9b..c08a0f2 100644
--- a/aws_integration/s3/setup.py
+++ b/aws_integration/s3/setup.py
@@ -4,6 +4,17 @@
 
 def after_migrate():
     """Create custom fields on File DocType for S3 integration."""
+    try:
+        _create_s3_custom_fields()
+    except Exception:
+        frappe.log_error(
+            title="aws_integration after_migrate failed",
+            message=frappe.get_traceback(),
+        )
+        frappe.logger("aws_integration").exception("after_migrate failed")
+
+
+def _create_s3_custom_fields():
     custom_fields = {
         "File": [
             {