diff --git a/s3torchconnector/pyproject.toml b/s3torchconnector/pyproject.toml
index 2f1d47ba..c78f58e0 100644
--- a/s3torchconnector/pyproject.toml
+++ b/s3torchconnector/pyproject.toml
@@ -44,7 +44,8 @@ e2e = [
 ]
 
 lightning = [
-    "lightning >= 2.0"
+    "lightning >= 2.0",
+    "s3fs"
 ]
 
 [tool.setuptools.packages]