From bbf399ef92215d221845c90ea4a434140e843031 Mon Sep 17 00:00:00 2001
From: Ben Bredesen <ben.bredesen@gmail.com>
Date: Thu, 13 Apr 2023 13:51:13 -0500
Subject: [PATCH] Fix for #43, set cat on every value

---
 feat/feature.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/feat/feature.go b/feat/feature.go
index 501c68e..d12c9cb 100644
--- a/feat/feature.go
+++ b/feat/feature.go
@@ -89,16 +89,20 @@ func (f *Feature) FilterByCategory() map[def.TypeCategory]*Feature {
 
 	// Stuff all the values, segmented first by category then by type, into the new Feature
 	// Lots of maps to make...
-	for _, vr := range f.ResolvedValues {
+	for k, vr := range f.ResolvedValues {
+		_ = k
 		// Default category reset before starting the inner loop
-		cat := def.CatExten
+		cat := def.CatNone
 
 		for valName, valDef := range vr {
 			if valDef.ResolvedType() != nil {
 				cat = valDef.ResolvedType().Category()
+			} else {
+				cat = def.CatExten
 			}
-			tmp := rval[cat]
-			if tmp == nil {
+
+			_, found := rval[cat]
+			if !found {
 				rval[cat] = NewFeature()
 			}