Fix SchemaTreeNamespace population via augment