diff --git a/skyvern/__init__.py b/skyvern/__init__.py
index c94a11fa..75f22af0 100644
--- a/skyvern/__init__.py
+++ b/skyvern/__init__.py
@@ -44,6 +44,7 @@ from skyvern.services.script_service import (  # noqa: E402
     run_script,  # noqa: E402
     run_task,  # noqa: E402
     send_email,  # noqa: E402
+    upload_file,  # noqa: E402
     wait,  # noqa: E402
 )  # noqa: E402
 
@@ -68,6 +69,7 @@ __all__ = [
     "run_task",
     "send_email",
     "setup",
+    "upload_file",
     "wait",
     "workflow",
 ]
diff --git a/skyvern/services/script_service.py b/skyvern/services/script_service.py
index 2013e335..67833f33 100644
--- a/skyvern/services/script_service.py
+++ b/skyvern/services/script_service.py
@@ -1479,7 +1479,7 @@ async def upload_file(
         label=block_validation_output.label,
         output_parameter=block_validation_output.output_parameter,
         parameters=parameters or [],
-        storage_type=storage_type,
+        storage_type=FileStorageType(storage_type),
         s3_bucket=s3_bucket,
         aws_access_key_id=aws_access_key_id,
         aws_secret_access_key=aws_secret_access_key,