diff --git a/cmd/yaml_test.go b/cmd/yaml_test.go
index 78b3101..4c7d67f 100644
--- a/cmd/yaml_test.go
+++ b/cmd/yaml_test.go
@@ -77,9 +77,15 @@
 		t.Errorf("Expected to find a recursive dependency")
 	}
 
-	imp := cfg.Imports[1]
-	if imp.Name != "github.com/Masterminds/convert" {
-		t.Errorf("Expected the convert package, got %s", imp.Name)
+	var imp *Dependency
+	for _, d := range cfg.Imports {
+		if d.Name == "github.com/Masterminds/convert" {
+			imp = d
+		}
+	}
+
+	if imp == nil {
+		t.Errorf("Expected the convert package, got nothing")
 	}
 
 	if len(imp.Subpackages) != 3 {