Add a check for the production server to the runner
Modify how total_nodes is used to avoid divide by 0 errors