diff --git a/arc_finetuning_st/streamlit/controller.py b/arc_finetuning_st/streamlit/controller.py
index df22158..41f1bba 100644
--- a/arc_finetuning_st/streamlit/controller.py
+++ b/arc_finetuning_st/streamlit/controller.py
@@ -157,7 +157,10 @@ def saved_finetuning_examples(self) -> List[str]:
 
     @property
     def task_file_names(self) -> List[str]:
-        return listdir(self._data_path)
+        try:
+            return listdir(self._data_path)
+        except FileNotFoundError:
+            return []
 
     def radio_format_task_name(self, selected_task: str) -> str:
         if selected_task in self.saved_finetuning_examples: