Fix health check