Fix stage failures. We need to wait for results to load