Now that I am thinking it through... I can actually get rid of that explicit wait.  Since the function is being used in a wait_for, there is an implicit wait of up to 5 seconds.  Thus, I don't need the explicit wait since there is a nice bound.