Add an await timeout