Fix failing build by adding timeout to tests