Make the health checker just take callables