tests: fix import order

Missing gofmt so much.