diff --git a/src/build/ast/Count.go b/src/build/ast/Count.go index 4c5e40e..7ef7b18 100644 --- a/src/build/ast/Count.go +++ b/src/build/ast/Count.go @@ -8,6 +8,9 @@ func Count(body AST, buffer []byte, kind token.Kind, name string) uint8 { for _, node := range body { switch node := node.(type) { + case *Assert: + count += node.Condition.Count(buffer, kind, name) + case *Assign: count += node.Expression.Count(buffer, kind, name)