Fixing null check when setting up probes