diff --git a/reclass/storage/yaml_git/__init__.py b/reclass/storage/yaml_git/__init__.py
index b08d4f9..ec78954 100644
--- a/reclass/storage/yaml_git/__init__.py
+++ b/reclass/storage/yaml_git/__init__.py
@@ -69,7 +69,7 @@
         return files
 
     def files_in_branch(self, branch):
-        tree = self.repo.revparse_single('master').tree
+        tree = self.repo.revparse_single(branch).tree
         return self.files_in_tree(tree, '')
 
     def files_in_repo(self):