Update timeout for tests