Changed how the #awaiting_doc method works. It was failing to find the button properly.