Fixed ordering problem with yaml_test.
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 {
