Added timeout in test