Set timeout on tests