Revert "Remove `WaitForResult` from two failing tests"

This reverts commit f0842409cde194ce2102ff2baf3de821cb6ac9d5.