A bug was introduced recently due to use of an undeclared variable. We need to add a unit test for this function, and fix this error.

NameError: name 'total_wait_seconds' is not defined