ForEach review and Commit.Tree err return

1 files changed, 6 insertions, 2 deletions

diff --git a/tree_test.go b/tree_test.go
index 92ae156..e27d4ad 100644
--- a/tree_test.go
+++ b/tree_test.go
@@ -188,7 +188,9 @@ func (s *SuiteTree) TestFile(c *C) {
 		commit, err := s.repos[t.repo].Commit(core.NewHash(t.commit))
 		c.Assert(err, IsNil, Commentf("subtest %d: %v (%s)", i, err, t.commit))
 
-		tree := commit.Tree()
+		tree, err := commit.Tree()
+		c.Assert(err, IsNil)
+
 		file, err := tree.File(t.path)
 		found := err == nil
 
@@ -260,7 +262,9 @@ func (s *SuiteTree) TestFiles(c *C) {
 		commit, err := s.repos[t.repo].Commit(core.NewHash(t.commit))
 		c.Assert(err, IsNil, Commentf("subtest %d: %v (%s)", i, err, t.commit))
 
-		tree := commit.Tree()
+		tree, err := commit.Tree()
+		c.Assert(err, IsNil)
+
 		var output []string
 		iter := tree.Files()
 		defer iter.Close()