cleanup import
test: Adjust capacity units
gofmt