make it possible to handle errors

1 files changed, 2 insertions, 2 deletions

diff --git a/reference_test.go b/reference_test.go
index ffa9f35..c8c7262 100644
--- a/reference_test.go
+++ b/reference_test.go
@@ -111,7 +111,7 @@ func TestIterator(t *testing.T) {
 		list = append(list, name)
 		name, err = iter.NextName()
 	}
-	if err != ErrIterOver {
+	if !IsErrorCode(err, ErrIterOver) {
 		t.Fatal("Iteration not over")
 	}
 
@@ -127,7 +127,7 @@ func TestIterator(t *testing.T) {
 		count++
 		_, err = iter.Next()
 	}
-	if err != ErrIterOver {
+	if !IsErrorCode(err, ErrIterOver) {
 		t.Fatal("Iteration not over")
 	}