Fix imports in `health_endpoint_test.go`