diff --git a/streamlit/interface_2.py b/streamlit/interface_2.py
index 65fd044..7bc7afc 100644
--- a/streamlit/interface_2.py
+++ b/streamlit/interface_2.py
@@ -95,7 +95,7 @@ s3_client = boto3.client('s3',
     region_name="us-east-2"
 )
 bucket = 'doczy-dev-infra-textract'
-objects = s3_client.list_objects_v2(Bucket=bucket, Prefix="training-data/contract-text-file/")
+objects = s3_client.list_objects_v2(Bucket=bucket, Prefix="training-data/")
 file_list = []
 for obj in objects['Contents']:
     if not obj['Key'].endswith('/'):
diff --git a/streamlit/interface_3.py b/streamlit/interface_3.py
index 102c221..711bf34 100644
--- a/streamlit/interface_3.py
+++ b/streamlit/interface_3.py
@@ -195,7 +195,7 @@ if st.session_state.user_info['mail'] in user_list:
         region_name="us-east-2"
     )
     bucket = 'doczy-dev-infra-textract'
-    objects = s3_client.list_objects_v2(Bucket=bucket, Prefix="training-data/contract-text-file/")
+    objects = s3_client.list_objects_v2(Bucket=bucket, Prefix="training-data/")
     file_list = []
     for obj in objects['Contents']:
         if not obj['Key'].endswith('/'):