diff --git a/opensfm/dataset.py b/opensfm/dataset.py
index 12af6a344..6b9c9d3dc 100644
--- a/opensfm/dataset.py
+++ b/opensfm/dataset.py
@@ -54,7 +54,10 @@ def __init__(self, data_path: str, io_handler=io.IoFilesystemDefault) -> None:
         self.load_mask_list()
 
     def _config_file(self) -> str:
-        return os.path.join(self.data_path, "config.yaml")
+        yaml_path = os.path.join(self.data_path, "config.yaml")
+        if self.io_handler.isfile(yaml_path):
+            return yaml_path
+        return os.path.join(self.data_path, "config.yml")
 
     def load_config(self) -> None:
         config_file_path = self._config_file()